随机森林的混淆矩阵（来自Scikit-学习）

Question

我正在为两个班级（0或1）执行随机森林分类，而且我经常得到混淆矩阵＆＃34;评估模型。

但我总是不确定哪一行是哪一类。 在这种情况下，我知道我可以告诉你的每个班级的数字。

你能告诉我简单的方法吗？

Answer 1

这在documentation：

中说明

  

标签：数组，形状= [n_classes]，可选

     

要编制索引的标签列表   矩阵。这可用于重新排序或选择标签的子集。   如果没有给出，那些在y_true或y_pred中至少出现一次的那些   按排序顺序使用。

这意味着这些类按升序排列，除非您通过提供标签列表（例如， confusion_matrix(y_true, y_pred, labels=[1, 0])将交换类。

元素[0,0]是0类中正确分类的数量，元素[1,1]是1类中正确分类的数量。