fklearn icon indicating copy to clipboard operation
fklearn copied to clipboard

Allow fit params for CatBoost

Open ValeriaGomes opened this issue 5 years ago • 0 comments

Describe the feature and the current state.

  • CatBoost has several parameters inside the fit method. One possible solution is to add a fit_params into the CatBoost function in FkLearn for increased flexibility.
  • This improvement would have to be done for both classification and regression.

This is important especially for CatBoost because the parameter that allows it to treat categorical features byitself is inside the fit method, and this is the main differential in this library.

Will this change a current behavior? How?

We would be able to configure the fit function using FkLearn.

Additional Information

https://catboost.ai/docs/concepts/python-reference_catboost_fit.html

ValeriaGomes avatar Jan 13 '20 17:01 ValeriaGomes