scikit-learn - 如何绘制仅与一门课程相关的分数的学习曲线

Question

我正在使用二元分类（2 类：A 和 B），我对绘制学习曲线感兴趣，但只与其中一个类（例如 B）相关，这是罕见的类。这可以用 scikit-learn 完成吗？到目前为止，我绘制的是 F1，这是 2 个班级的 F1 平均值，但我看不到仅绘制每个班级的分数的方法。

Answer 1

一个班级的 F1 分数并没有真正的意义。对 1 或“正类”（可能是罕见的）进行预测的方式发生任何变化，显然会改变您预测 0 的频率。

可能更有用的是绘制稀有类的召回率以及它如何随着您拟合模型而增加。您还可以绘制假阴性的发生情况（模型预测为 0 但应该预测罕见的正类的情况）。这些将显示随着模型的改进，对单个类的预测如何变得更加精确。