Imbalanced Learn Pipeline & Testing

uclamii / model_tuner

A library to tune the hyperparameters of common ML models. Supports calibration and custom pipelines.

Apache License 2.0

3 stars 0 forks source link

Closed elemets closed 6 months ago

elemets commented 6 months ago

[x] Change out pipeline if imbalanced being used
[ ] Unit testing (assertions that over sampling / under sampling the class balance has changed and minority and majority is 50/50)
[ ] Random seed match
[ ] Test cases: (All of these for Multi Label, Regression and Binary Classification)
[ ] XGBoost
[ ] Logistic Regression
[ ] AutoKeras
[ ] Random Forest

elemets commented 6 months ago

[] ensure imbalanced learn is work in XGBoost early stopping case