如何绘制只与一门课相关的分数的学习曲线



我正在使用一个二元分类(2类:a和B),我对绘制学习曲线感兴趣,但只与其中一个类有关,比如B,这是罕见的类。scikit-learn能做到吗?到目前为止我画的是F1,这是两个班级的F1平均值但我没有办法单独画出每个班级的分数

单个类的F1分数没有真正意义。你对1或"正类"(可能是罕见的一类)的预测方式的任何改变,都明显改变了你预测0的频率。

可能更有用的是绘制罕见类的召回率,以及当你拟合模型时它是如何增加的。您还可以绘制假阴性的出现情况(模型预测为0,但应该预测罕见的阳性类别)。这将表明,随着模型的改进,对单个类别的预测如何变得更加精确。

相关内容

  • 没有找到相关文章

最新更新