Skip to content

MilesCranmer/bnn_chaos_model

master
Switch branches/tags

Name already in use

A tag already exists with the provided branch name. Many Git commands accept both tag and branch names, so creating this branch may cause unexpected behavior. Are you sure you want to create this branch?
Code

Latest commit

 

Git stats

Files

Permalink
Failed to load latest commit information.
Type
Name
Latest commit message
Commit time
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

A Bayesian neural network predicts the dissolution of compact planetary systems

This repository is the official implementation of the paper https://arxiv.org/abs/2101.04117, containing training code, evaluation scripts, and figure generation.

For an easy-to-use API implementing a trained version of this model, check out spock!

Running

To train the model, download the data into the /data folder from this Globus link.

Run the script train.sh, which will train the model 30 times from different seeds.

To generate the figures, edit the figures/generate.sh script to have true instead of false for any figure you'd like to generate. Then, execute that script in the folder.

Requirements

One can install a working conda environment using create_env.sh.

This file also contains a list of packages required by this model.

About

Model and training code for Bayesian neural network for compact planetary instability

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published