Bayesian update train val test fix - 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

Bayesian update train val test fix #65

Closed elemets closed 2 weeks ago

elemets commented 2 weeks ago

Bayesian / KFold Fixes

Fixed kfold and bayesian .fit() methods
A fitted model is now returned and can be used
Bayesian parameters can be set as shown in the bayesiantest.py

train_val_test_split fixes

Remove unnecessary comments
Moved to outside of the class
Updated stratify_y to is not None to stop bugs