python/sklearnのあるクラスの確率値の列を取得する

Question

sklearnでrandomforestを使用してバイナリ分類を行うとき。 予測クラスを取得するのにclf.predict(X)を使うことができます。 そして、私はclf.predict_proba(X)を使用する場合、私は配列は次のようになり得る：

私は最初の列は、予測の確率を示すと思いますか？ クラスが1になる可能性のある列を取得するにはどうすればよいですか？ RandomForest.predict_probaドキュメントから

Answer 1

：
p：[...]入力サンプルのクラス確率

predict_proba(X) 

はX.

戻りのためのクラス確率を予測します。クラスの の順番は、属性classes_の順序に対応します。

あなたは、clf.classes_属性を見て、あなたのクラス1が表示されますどのような指標で確認した後、同じようにその確率をアクセスすることができます。iはclf.classes_でクラス1の指標である

prob_class_1 = clf.predict_proba(X)[:, i] 

を。