在CatBoost中测量AUPRC

时间:2019-03-03 00:36:25

标签: machine-learning precision-recall catboost

我想在catboost中测量精确召回曲线(AUPRC)曲线下的面积,但是CatBoostClassifier并没有AUPRC作为评估指标。任何建议都可以帮助我进行测量此性能指标将不胜感激。

这是我使用的代码:

model = CatBoostClassifier(
custom_loss=['Accuracy','Precision','Recall','F1','AUC'],
random_seed=42,
logging_level='Silent')
model.fit(
X_train, Y_train,
eval_set=(X_test, Y_test), plot=True)

1 个答案:

答案 0 :(得分:0)

Catboost具有MAP度量,代表平均平均精度。与AUCPR几乎相同。 reference article很好地解释了原因。