predict_proba メソッドは、各クラス（商品サイズ）に対する予測確率を返す？

2024年6月8日2024年6月17日

predict_proba メソッドは、各クラス（商品サイズ）に対する予測確率を返します。具体的には、各サンプル（テストデータ）について、それが各クラスに属する確率を返します。各行は1つのサンプルに対応し、各列は各クラスに対応します。

例えば、3つのクラス（商品サイズ）がある場合、出力される行列は次のような形式になります。

css
[[0.1, 0.6, 0.3],
 [0.4, 0.3, 0.3],
 [0.2, 0.2, 0.6],
 ...
]

各行の合計は1になります。各行の各列の値は、対応する商品サイズに属する確率を表します。この例では、1つ目のサンプルはクラス2に、2つ目のサンプルはクラス1に、3つ目のサンプルはクラス3に属する確率が高いと予測されています。