随机森林的混淆矩阵(来自Scikit-learn)

confusion matrix of random forest ( from Scikit- learn)

我正在对两个 classes(0 或 1)执行随机森林 class化,我经常 "confusion matrix" 来评估模型。

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

能告诉我简单的方法吗?

这在documentation中有说明:

labels : array, shape = [n_classes], optional

List of labels to index the matrix. This may be used to reorder or select a subset of labels. If none is given, those that appear at least once in y_true or y_pred are used in sorted order.

这意味着 classes 是按升序排列的,除非您通过提供标签列表来进行 sklearn,例如confusion_matrix(y_true, y_pred, labels=[1, 0]) 会交换 classes。

元素 [0,0] 是 0 class 中正确的 class 化的数量,元素 [1,1] 是正确的 class 化的数量1 class.