Stability of Planetary Orbital Configurations Klassifier


Keywords
astronomy, astrophysics, exoplanets, stability
License
GPL-2.0+
Install
pip install spock==1.6.0

Documentation

SPOCK 🖖

Stability of Planetary Orbital Configurations Klassifier

image image image image image image image image

Documentation

The SPOCK package incorporates several machine learning and analytical tools for estimating the stability of compact planetary configurations. All estimators use a common API to facilitate comparisons between them and with N-body integrations.

Quickstart

Let's predict the probability that a given 3-planet system is stable past 1 billion orbits with the XGBoost-based classifier of Tamayo et al., 2020.

import rebound
from spock import FeatureClassifier
feature_model = FeatureClassifier()

sim = rebound.Simulation()
sim.add(m=1.)
sim.add(m=1.e-5, P=1., e=0.03, pomega=2., l=0.5)
sim.add(m=1.e-5, P=1.2, e=0.03, pomega=3., l=3.)
sim.add(m=1.e-5, P=1.5, e=0.03, pomega=1.5, l=2.)
sim.move_to_com()

print(feature_model.predict_stable(sim))
# >>> 0.06591137

That model provides a simple scalar probability of stability over a billion orbits. We can instead estimate its median expected instability time using the deep regressor from Cranmer et al., 2021.

from spock import DeepRegressor
deep_model = DeepRegressor()

median, lower, upper = deep_model.predict_instability_time(sim, samples=10000)
print(int(median))
# >>> 242570.1378387966

The returned time is expressed in the time units used in setting up the REBOUND Simulation above. Since we set the innermost planet orbit to unity, this corresponds to 242570 innermost planet orbits.

Finally, we can compare these results to the semi-analytic criterion of Tamayo et al., 2021 for how likely the configuration is to be dynamically chaotic. . This is not a one-to-one comparison, but configurations that are chaotic through two-body MMR overlap are generally unstable on long timescales (see paper and examples).

from spock import AnalyticalClassifier
analytical_model = AnalyticalClassifier()

print(analytical_model.predict_stable(sim))
# >>> 0.0

To match up with the above classifiers, the analytical classifier returns the probability the configuration is regular, i.e., not chaotic. A probability of zero therefore corresponds to confidently chaotic. See this example for more information about the analytical model.

Examples

Colab tutorial for the deep regressor.

The example notebooks contain many additional examples: jupyter_examples/.

Installation

SPOCK is compatible with both Linux and Mac. SPOCK relies on XGBoost, which has installation issues with OpenMP on Mac OSX. If you have problems (dmlc/xgboost#4477), the easiest way is probably to install homebrew, and:

brew install libomp

The most straightforward way to avoid any version conflicts is to download the Anaconda Python distribution and make a separate conda environment.

Here we create we create a new conda environment called spock and install all the required dependencies

conda create -q --name spock -c pytorch -c conda-forge python=3.7 numpy scipy pandas scikit-learn matplotlib torchvision pytorch xgboost rebound einops jupyter pytorch-lightning ipython h5py
conda activate spock
pip install spock

Each time you want to use spock you will first have to activate this spock conda environment (google conda environments).