开始使用免费开始使用

混淆矩阵

使用 scikit-learn 的 confusion_matrix() 函数,您可以轻松生成分类器的混淆矩阵,从而更细致地理解模型表现。它需要两个参数:测试集的真实标签 y_test,以及模型的预测标签。

上一练习中随机森林分类器的预测标签已存储在 y_pred,并通过以下代码计算:

y_pred = clf.predict(X_test)

重要提示:sklearn 默认按如下方式计算混淆矩阵:

Screenshot 2019-05-13 05.59.04.png

请注意,坐标轴与您在视频中看到的是相反的。各项指标本身不变,但解读表格时请记住这一点。

本练习是课程的一部分

营销分析:用 Python 预测客户流失

查看课程

交互式实操练习

通过完成这段示例代码来试试这个练习。

# Import confusion_matrix
编辑并运行代码