ExoBoost

Using existing radial velocity data to identify exoplanet companions with XGBoost

Code Author

Rohan S. Nagavardhan: @rndev2017

Directories

src/

• contains the XGBoost Classifier
• contains the script used to label my data set
• contains the Star class
  • plot Radial Velocity Plots
  • plot phased Radial Velocity Plots
  • plot periodograms
  • fit phased radial velocity curves

plots/

• contains images that are relavent to the model
• contains images that are relavent to the exoplanet data

sim/

• contains code for creating synthetic radial velocity data

data/

• contains the data used to train the XGBoost Classifier
• formatted in .csv format

Dependencies

• XGBoost (instructions)
• Pandas (instructions)
• NumPy (instructions)
• SciPy (instructions)
• Astropy (instructions)
• Matplotlib (instructions)
• Scikit-Learn - (instructions)
• Seaborn - (instructions)
• Astroquery - (instructions)

Future Plans

I have been given the oppurtunity to present this work at LISEF and various science research fairs in the near future as part of my science research class. Additionally, I am considering submitting this work to a journal for publication. As for the project itself, I will be working on improving the model by creating a simulated dataset to add to the existing dataset of real exoplanet data. Furthermore, I am thinking about using a transfer learning approach by repurposing AstroNet for exoplanet companion identification with radial velocity. Keep following this directory for updates on this project.