scikit-learn中predict_proba的输出

Question

假设我有一个数据样本，其中有两个类别标记为 0 和 1。当我运行 时output = clf.predict_proba(X_input)， 中的每一行output由 2 列组成，对应于每个类别的概率。

第一列代表 0 类还是 1 类的概率？GradientBoostingClassier的predict_proba方法说：

“输入样本的类别概率。类别的顺序与属性classes_中的顺序相对应。”

这是否意味着数据样本的第一个元素（0 或 1）对应于 输出中的第一列predict_proba？

Answer 1

通常，分类器会有一个名为该属性的属性，classes_该属性将在拟合时填充并存储类。方法输出的顺序predict_proba将与此属性中的顺序相同。

例如：

nb = MultinomialNM()
nb.fit(some_gender_data)
nb.classes_
array(['F', 'M'], dtype='<U1')

据我所知，sklearn 中的所有分类器一旦拟合都具有此属性。