首页 > 解决方案 > 如何使用 AdaBoostClassifier 中的概率获得预测类别?

问题描述

我尝试使用预测概率来获得预测的类,但是分别给定类 [-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

标签: pythonscikit-learnprobabilityadaboost

解决方案


推荐阅读