Kfold threshold tuning fix

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 4 months ago

elemets commented 4 months ago

KFold threshold tuning bug fix
- Changed the method to use predictproba outside of function so that it works without doing a new prediction every-time.

lshpaner commented 4 months ago

looks good; predict_proba working with returning the correct labels.