CatBoostClassifier - AUC度量

Question

我对CatBoostClassifier有疑问。

params = {
'loss_function' : 'Logloss',
'eval_metric' : 'AUC',
'verbose' : 200,
'random_seed' : 42,
'custom_metric' : 'AUC:hints=skip_train~false'
}

cbc = CatBoostClassifier(**params)
cbc.fit(x_tr, y_tr,
        eval_set = (x_te, y_te),
        use_best_model = True,
        plot = True
        );

predictions = cbc.predict(x_te)

模型结果：

bestTest = 0.6786987522

但当我尝试：

 from sklearn import metrics
 auc = metrics.roc_auc_score(y_te, predictions)
 auc

我得到了0.5631684491978609的结果。为什么这个结果不同？第一和第二结果意味着什么？哪一个是我的cbc模型的最终度量？

Answer 1

好吧，我找到解决办法了。我应该使用：

predictions = cbc.predict_proba(x_te) 

而不是

predictions = cbc.predict(x_te)

现在我有了同样的结果。