Add CatBoostRegressor to MapieQuantileRegressor class #217
Labels
Enhancement
Type: enhancement (new feature or request)
Good first issue
Easy issue to start to contribute to MAPIE
Needs decision
The MAPIE team is deciding what to do next.
Regression
Related to regression (excluding time series)
Source: contributors
Proposed by contributors.
Milestone
Is your feature request related to a problem? Please describe.
Currently, I'm using CatBoost regressor for a regression problem, where this model seems to give best performance. Right now, I'm looking to produce prediction intervals, and would like to stick with the same model, and at the same time, make use of MAPIE and not make separate models in code.
Describe the solution you'd like
Use CatBoost regressor with CQR.
Describe alternatives you've considered
Solution is not using MAPIE and create 2 additional models and follow same approach to calculate q, and then form the sets. Other is to modify code base and add CatBoost regressor support, but not really good at coding, still understanding quantile_regression.py file, to see how to add CatBoost, since I think, the only way to use quantile loss function and alpha, is by setting them both values in the same string like this:
model = catboost.CatBoostRegressor(loss_function='Quantile:alpha=0.95', ...)
*I can maybe take a look at it and work on it, but will take time. It would be a good first contribution to do for me :)
Thanks in advance,
The text was updated successfully, but these errors were encountered: