如何使用AdaBoostClassifier中的概率获得预测的类别?

时间:2019-04-30 08:03:25

标签: python scikit-learn probability adaboost

我尝试使用预测概率来获得预测类别,但是在给定类别分别为[-1,1]的情况下,准确性会急剧变化。

这是我用来将概率转换为预测类的代码。

y_pred_prob = []
for i in Y_prob:
    if i[0] > i[1]:
        y_pred_prob.append(-1)
    else:
        y_pred_prob.append(1)

当我使用预测类时,准确性为0.8983,但是当我使用y_pred_prob时准确性为0.6241

0 个答案:

没有答案