MLBox icon indicating copy to clipboard operation
MLBox copied to clipboard

Published 20 hours ago • verified publisher icon AxeldeRomblay

Add CatBoost

Open annaveronika opened this issue 4 years ago • 5 comments

It would be nice to add CatBoost, one of three most popular gbdt libraries (https://github.com/catboost/catboost, https://catboost.ai/)

annaveronika avatar Apr 30 '20 09:04 annaveronika

Hello @annaveronika, yes I know that one... Do you have a benchmark between LightGBM & Catboost ? Because there is already a gbdt algo in MLBox... Also is it easy to install ?

AxeldeRomblay avatar Apr 30 '20 10:04 AxeldeRomblay

pip install catboost should work The difference in resulting metrics between xgboost, catboost and lightgbm will depend on the dataset, there is no single winner for datasets with only numeric features. If you run the algorithm with default parameters, catboost will usually win, if you run with parameter tuning, you might get any of them as a winner. If you have categorical features or text features catboost will usually win.

annaveronika avatar Apr 30 '20 13:04 annaveronika

Ok. Maybe you can try to send me a PR (code + tests) if you are comfortable with this... otherwise I will try to have a look.

AxeldeRomblay avatar May 04 '20 08:05 AxeldeRomblay

It would be great if you would do it when you have time. I don't plan for pr-s, sorry...

annaveronika avatar May 04 '20 09:05 annaveronika

@AxeldeRomblay Can I raise a PR for this?

divyaprabha123 avatar May 26 '20 14:05 divyaprabha123