merge to main - Githubissues

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

merge to main #1

Closed lshpaner closed 5 months ago

lshpaner commented 5 months ago

updated s/t calibration works with Python > 3.7
added ability to parse in SelectKbest()
added flexibility for updating calibration method (i.e, isotonic)
best score print post calibration
added early stopping for XGBoost
added ability to extract train, valid, test sets after split
added classification report, confusion matrix report, KFold report
added AutoKeras wrapper to work with this implementation
multiclass/multilabel
regression
custom scoring
threshold tuning for KFold

elemets commented 5 months ago

Need to look at comments and remove TODOs that have been completed.