Python的predict_proba类标识

Question

假设我的标记数据有两个类1和0.当我在测试集上运行predict_proba时，它返回一个包含两列的数组。哪个列对应哪个类？

Answer 1

第0列对应于第0类，第1列对应于第1类。

Answer 2

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