Fixing bug related to n_estimators not being used

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

elemets commented 3 months ago

Extract the number of estimators out of the xgboost model in early stopping case
Add it back into the parameters to be used for fitting