Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add support for CatBoostRegressor #1429

Open
krishanbhasin-gc opened this issue Oct 3, 2023 · 0 comments
Open

Add support for CatBoostRegressor #1429

krishanbhasin-gc opened this issue Oct 3, 2023 · 0 comments

Comments

@krishanbhasin-gc
Copy link
Contributor

Following #1403, it would be great to also support the CatBoostRegressor (and subsequently CatBoostRanker) model types.

From the linked PR:

Q:

Looking ahead to adding support for the Regressor and Ranker models, given catboost/catboost#2504 we can't use the same approach as XGBoost (using a try-catch).
It might be worth creating 3 separate MLModel child classes, one for each of Classifier, Regressor, and Ranker instead. Could this fit the design approach of the project?

A:

On what you mention around classifiers vs regressors, I think we should land this one as a first iteration and then open an issue to add support for extra types of CatBoost models. It's a shame the framework itself doesn't give us any way to differentiate between artefacts - in that case, we may need to add some extra config field to the model-settings.json to select between model types.

@krishanbhasin-gc krishanbhasin-gc changed the title Add support for CatBoostRegressor Add support for CatBoostRegressor Oct 3, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant