CatBoost回归中的预测概率



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

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

model = CatBoostRegressor(loss_function='RMSE', silent=True)
model.fit(x_train, y_train)
y_pred = model.predict(x_test, prediction_type='Probability')

最新更新