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

Catboost support #36

Closed elemets closed 1 month ago

elemets commented 1 month ago
elemets commented 1 month ago

This is relevant to issue #34

lshpaner commented 1 month ago

Thank you for this solid contribution!

The bug fix for the stratify_cols issue was much needed, especially addressing the length mismatch by correcting the use of y_valid_test instead of y.

Adding CatBoost support with early stopping is a great enhancement, and I appreciate the robust handling of the best iteration extraction with the try-except block.

The fix for the ambiguity related to the truth value of a list/dataframe is also crucial for improving reliability. This PR significantly improves the functionality and stability of the codebase—nicely done!