如何预测我们的模型95%确定的值？

Question

我有5节课。我建立了一个XGBoost分类模型，并使用model.predict(test)来预测测试数据集的类。在我的模型所预测的所有值中，我只想知道那些我的模型超过95%的值，确信预测值是正确的。我的意思是，我只希望那些我的模型非常有信心预测的预测。我怎样才能找到那些预言？

Answer 1

看看predict_proba的方法XGBClassifier班级，它将给出每个类的概率，而不仅仅是预测的类。然后，如果概率高于要设置的阈值(在本例中为0.95)，则只能使用这些概率来选择概率最高的类。