This repository contains the trained machine learning models and python notebooks for the paper Machine-learning inference of the interior structure of low-mass exoplanets (Baumeister et al. 2020).

This project requires Python 3.

• keras = 2.2.4
• numpy = 1.18.0
• scipy >= 1.2.0
• matplotlib >= 3.0.2
• tensorflow = 1.15.2
• tensorflow-probability = 0.7.0
• ipywidgets >= 7.4.2
• joblib >= 0.13.2
• scikit-learn = 0.22.1

conda env create -f requirements.yml

Activate with

conda activate tf1.15

pip install -r requirements.txt

• MDN_exoplanets.ipynb contains all the code to load the trained MDN models and predict the distribution of possible interior structures of a planet.
• The mdn directory contains the MDN layer code adopted from https://github.com/cpmpercussion/keras-mdn-layer.
• The models directory contains data scalers and the MDN models trained either with mass and radius of the planet, or with mass, radius, and k₂.