MLServer icon indicating copy to clipboard operation
MLServer copied to clipboard

Add support for `CatBoostRegressor`

Open krishanbhasin-gc opened this issue 2 years ago • 0 comments

Following https://github.com/SeldonIO/MLServer/pull/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 https://github.com/catboost/catboost/issues/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 avatar Oct 03 '23 13:10 krishanbhasin-gc