search

aporia-ai / mlnotify

🔔 No need to keep checking your training - just one import line and you'll know the second it's done.
https://mlnotify.aporia.com
MIT License
344 stars 19 forks source link

Add catboost #44

Closed PragyanSubedi closed 3 years ago

PragyanSubedi commented 3 years ago

I've added in two commonly used Catboost models with this PR (CatBoostRegressorand CatBoostClassifier). Please take a look at the changes and let me know if any changes are needed!

SnirShechter commented 3 years ago

@PragyanSubedi Looks good. Can you add tests?

PragyanSubedi commented 3 years ago

@PragyanSubedi Looks good. Can you add tests?

I've added it now. Please have a look. @SnirShechter

SnirShechter commented 3 years ago

@PragyanSubedi Sorry, I probably missed your comment.

Merged. Thanks!