Training Models

Train the three bundled classifiers using the dataset in data/.

RandomForest

from exobengal.exobengal import DetectExoplanet

detector = DetectExoplanet()
detector.train_random_forest(data_path="data/cumulative_2025.09.20_12.15.37.csv")

CNN

from exobengal.exobengal import DetectExoplanet

detector = DetectExoplanet()
detector.train_cnn(data_path="data/cumulative_2025.09.20_12.15.37.csv")

kNN

from exobengal.exobengal import DetectExoplanet

detector = DetectExoplanet()
detector.train_knn(data_path="data/cumulative_2025.09.20_12.15.37.csv")

API Reference

Prediction