CODEWITHSUNDEEP
catboostcatboostregressor
user308827
user308827

Reputation: 21961

Predicting probabilities in CatBoost regressor

Does CatBoost regressor have a method to predict the probabilities of each prediction? I see one for CatBoost classifier (https://catboost.ai/en/docs/concepts/python-reference_catboostclassifier_predict_proba) but not for regressor.

Upvotes: 3

Views: 2510

Answers (1)

user4718221
user4718221

Reputation: 606

There is no predict_proba method in the Catboost regressor, but you can specify the output type when you call predict on the trained model.

model = CatBoostRegressor(loss_function='RMSE', silent=True)
model.fit(x_train, y_train)

y_pred = model.predict(x_test, prediction_type='Probability')

Upvotes: 3

Related Questions