Skip to content

Machine learning models and utilities for exoplanet science.

License

Notifications You must be signed in to change notification settings

M-68/exoplanet-ml

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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

astronet/

  • 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.

astrowavenet/

  • A generative model for light curves.

light_curve/

  • 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.

tf_util/

  • Shared TensorFlow utilities.

third_party/

  • Utilities derived from third party code.

Setup

Required Packages

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.

About

Machine learning models and utilities for exoplanet science.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 80.9%
  • C++ 19.1%