Models & Artifacts
Directory: models/
- random_forest_classifier.pkl — RandomForestClassifier
- cnn_model.h5 — Keras dense model
- knn_model.pkl — KNeighborsClassifier
- scaler.pkl — StandardScaler
- imputer.pkl — SimpleImputer
Prediction (auto-load)
from exobengal.exobengal import DetectExoplanet
m = DetectExoplanet()
print(m.random_forest([365.0, 1.0, 288.0, 1.0, 4.44, 5778, 0.1, 5.0, 100.0]))
Retraining behavior
- Each
train_*
overwrites its model and writesscaler.pkl
andimputer.pkl
. - Move or rename artifacts beforehand if you want to keep previous versions.