Project: exoplanet-ml (GitHub Link)

exoplanet-ml-master
- AUTHORS
- exoplanet-ml
  - box_least_squares
    - BUILD
    - box_least_squares_test.cc
    - box_least_squares_impl_test.cc
    - bin_by_phase_test.cc
    - util.h
    - test_util.h
    - box_least_squares_impl.cc
    - box_least_squares.h
    - box_least_squares_impl.h
    - box_least_squares.proto
    - python
      - box_least_squares.clif
      - bin_by_phase_test.py
      - box_least_squares_test.py
      - bin_by_phase.clif
    - bin_by_phase.h
    - box_least_squares.cc
  - third_party
    - robust_mean
      - BUILD
      - LICENSE
      - robust_mean_test.py
      - test_data
        random_normal.py
        __init__.py
      - robust_mean.py
      - __init__.py
    - kepler_spline
      - BUILD
      - kepler_spline.py
      - kepler_spline_test.py
      - LICENSE
      - __init__.py
    - __init__.py
  - experimental
    - README.md
    - light_curve
      - BUILD
      - transit_model.py
    - beam
      - transit_search
        BUILD
        bls_fns.py
        beam_transit_search.py
        kepler_id.py
        beam_astronet_predict.py
        prediction_fns.py
        bls_scorer.py
      - light_curve
        BUILD
        transit_fns.py
  - astrowavenet
    - BUILD
    - astrowavenet_model.py
    - astrowavenet_model_test.py
    - util
      - BUILD
      - estimator_util.py
    - configurations.py
    - data
      - BUILD
      - synthetic_transits.py
      - kepler_light_curves.py
      - test_data
        test-dataset.tfrecord
      - synthetic_transit_maker_test.py
      - base_test.py
      - __init__.py
      - synthetic_transit_maker.py
      - base.py
    - __init__.py
    - README.md
    - trainer.py
  - light_curve
    - binning.py
    - BUILD
    - util.py
    - kepler_io_test.py
    - light_curve.proto
    - periodic_event_test.py
    - kepler_io.py
    - util_test.py
    - fast_ops
      - median.h
      - BUILD
      - median_test.cc
      - phase_fold.h
      - view_generator_test.cc
      - normalize.h
      - median_filter.h
      - view_generator.cc
      - normalize_test.cc
      - test_util.h
      - python
        view_generator.clif
        postproc.py
        median_filter_test.py
        view_generator_test.py
        phase_fold_test.py
        phase_fold.clif
        median_filter.clif
      - phase_fold.cc
      - median_filter.cc
      - median_filter_test.cc
      - normalize.cc
      - phase_fold_test.cc
      - view_generator.h
    - test_data
      - 0114
        011442793
        kplr011442793-2010009091648_llc.fits
    - binning_test.py
    - periodic_event.py
    - __init__.py
    - README.md
  - tf_util
    - BUILD
    - configdict_test.py
    - config_util_test.py
    - config_util.py
    - estimator_runner.py
    - example_util.py
    - __init__.py
    - example_util_test.py
    - configdict.py
  - WORKSPACE
  - astronet
    - BUILD
    - astro_fc_model
      - BUILD
      - astro_fc_model.py
      - configurations.py
      - astro_fc_model_test.py
      - __init__.py
    - astro_cnn_model
      - BUILD
      - astro_cnn_model.py
      - astro_cnn_model_test.py
      - configurations.py
      - __init__.py
    - evaluate.py
    - util
      - BUILD
      - estimator_util.py
      - __init__.py
    - models.py
    - predict.py
    - data
      - BUILD
      - generate_input_records.py
      - preprocess.py
      - generate_download_script.py
      - __init__.py
    - train.py
    - __init__.py
    - README.md
    - ops
      - testing.py
      - BUILD
      - input_ops_test.py
      - training.py
      - metrics_test.py
      - metrics.py
      - test_data
        test_dataset.tfrecord
      - __init__.py
      - dataset_ops_test.py
      - input_ops.py
      - dataset_ops.py
    - astro_model
      - BUILD
      - configurations.py
      - __init__.py
      - astro_model_test.py
      - astro_model.py
    - docs
  - .gitignore
  - beam
    - BUILD
    - utils.py
    - astrowavenet
      - BUILD
      - beam_prepare_inputs.py
      - beam_predict.py
      - beam_visualize.py
      - visualize_fns.py
      - embedding_fns.py
      - prediction_fns.py
      - process_light_curve.py
    - beam_sample_tfrecord.py
    - light_curve
      - BUILD
      - light_curve_fns.py
    - astronet
      - BUILD
      - beam_reshuffle.py
      - beam_prepare_embedding_inputs.py
      - beam_prepare_inputs.py
- LICENSE
- CONTRIBUTING.md
- README.md

Exoplanet ML

Machine learning models and utilities for exoplanet science.

Code Author

Chris Shallue: @cshallue

Walkthrough

You can jump straight to the AstroNet walkthrough.

Otherwise, click through to the desired directory as outlined below.

Directories

A neural network for identifying exoplanets in light curves. Contains code for:
- Downloading and preprocessing Kepler light curves.
- Building different types of neural network classification models.
- Training and evaluating a new model.
- Using a trained model to generate new predictions.

A generative model for light curves.

Utilities for operating on light curves. These include:
- Reading Kepler data from .fits files.
- Applying a median filter to smooth and normalize a light curve.
- Phase folding, splitting, removing periodic events, etc.
light_curve/fast_ops/ contains optimized C++ light curve operations.

Shared TensorFlow utilities.

Utilities derived from third party code.

Setup

Required Packages

TensorFlow (instructions)
Pandas (instructions)
NumPy (instructions)
SciPy (instructions)
AstroPy (instructions)
PyDl (instructions)
Bazel (instructions)
Abseil Python Common Libraries (instructions)

Run Unit Tests

Verify that all dependencies are satisfied by running the unit tests:

cd exoplanet-ml  # Bazel must run from a directory with a WORKSPACE file
bazel test astronet/... astrowavenet/... light_curve/... tf_util/... third_party/...

Citation

If you find this code useful, please cite our paper:

Shallue, C. J., & Vanderburg, A. (2018). Identifying Exoplanets with Deep Learning: A Five-planet Resonant Chain around Kepler-80 and an Eighth Planet around Kepler-90. The Astronomical Journal, 155(2), 94.

Full text available at The Astronomical Journal.

Disclaimer

This is not an official Google product.