Измерение AUPRC в CatBoost - PullRequest
Новая
       80

Измерение AUPRC в CatBoost

0 голосов
/ 03 марта 2019

Я хочу измерить площадь под кривой кривой точного отзыва (AUPRC) в catboost, но 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)
...