在 python/sklearn 中获取某个类的概率值列

Question

在 sklearn 中使用 randomforest 进行二元分类时。我知道我可以clf.predict(X)用来获得预测的课程。当我使用时clf.predict_proba(X)，我得到一个如下所示的数组：

我认为第一列表示预测的概率？我怎样才能得到一列，该列的概率为 1？

Answer 1

从RandomForest.predict_proba文档：

predict_proba(X)

预测 X 的类别概率。

返回::
p[...] 输入样本的类别概率。类的顺序对应于属性中的顺序classes_。

您可以查看clf.classes_属性，查看您的类1出现的索引，然后像这样访问概率：

prob_class_1 = clf.predict_proba(X)[:, i] 

哪里i是类指数1在clf.classes_。