2

SKLearn でロジスティック回帰を使用して、データを 5 つのクラスのいずれかに分類しています。データをトレーニングするために、観測Yのマトリックスと機能のマトリックスがありますX

場合によっては、私のマトリックスYにカテゴリ 3 の発言権がない場合があります。この場合、predict_proba(X)メソッドを呼び出すときに、3 番目のエントリが 0 である 5 つの確率のリストが必要です (カテゴリ 3 の観測がないため)。代わりに、この確率は単純に省略され、4 つの確率のリストが返されます。

これを行うためにロジスティック回帰オブジェクトを変更するにはどうすればよいですか?

4

2 に答える 2

1

モジュールを使用して、マルチクラス ラベルを見つけることができsklearn.preprocessingます。

参照: http://scikit-learn.org/stable/modules/preprocessing.html#label-binarization

于 2014-03-12T11:16:59.733 に答える