plot - scikit-learn - 1 つのクラスのみに関連するスコアの学習曲線をプロットする方法

Question

私はバイナリ分類 (2 つのクラス: A と B) を使用しており、学習曲線をプロットすることに興味がありますが、クラスの 1 つ、たとえば珍しいクラスである B との関係のみに関心があります。scikit-learnでそれを行うことは可能ですか? これまでにプロットしたのは F1 で、これは 2 つのクラスの F1 平均ですが、それぞれのスコアを個別にプロットする方法がわかりません。

Answer 1

1 つのクラスの F1 スコアはあまり意味がありません。1または「ポジティブクラス」（おそらくまれなもの）の予測を行う方法を変更すると、0を予測する頻度が明らかに変わります。

より役立つのは、まれなクラスの再現率と、モデルに適合するにつれて再現率がどのように増加するかをプロットすることです。また、偽陰性 (モデルが 0 を予測したが、まれな陽性クラスを予測する必要がある場合) の発生をプロットすることもできます。これらは、モデルが改善されるにつれて、個々のクラスの予測がどのように正確になるかを示します。