在 CatBoost 回归器中预测概率

Question

CatBoost 回归器是否有预测每个预测概率的方法？我看到一个用于 CatBoost 分类器 (https://catboost.ai/en/docs/concepts/python-reference_catboostclassifier_predict_proba) 但不是用于回归器。

Answer 1

Catboost 回归器中没有 predict_proba 方法，但您可以在训练模型上调用预测时指定输出类型。

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

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

在 CatBoost 回归器中预测概率

Predicting probabilities in CatBoost regressor

catboost

catboostregressor