Python predict_proba class 标识

Question

假设我标记的数据有两个 classes 1 和 0。当我 运行 predict_proba 在测试集上 returns 一个有两列的数组。哪一列对应于哪个 class ？

Answer 1

第0列对应class0，第1列对应class1.

Answer 2

您可以通过打印 类 和 print(estimator.classes_) 来检查。该数组将具有与输出相同的顺序。