scikit-learnの多クラス分類でのpredictについて

Question

scikit-learnのSVCを用いて.predict等で多クラス分類の答えを予測する際に、予測結果を一つのみ出力するだけでなく第2候補、第3候補など順位付けされた複数の予測の出力は可能でしょうか？また、それらの結果の確率分布をパーセンテージなどの数値で出力する事は可能でしょうか？
メソッドなどがあれば教えていただきたいです。ググり方でも大丈夫です。
漠然とした質問ですみませんよろしくお願いします。

Answer 1

下記メソッドをその目的に使うことができます。

property predict_proba
Compute probabilities of possible outcomes for samples in X.

The model need to have probability information computed at training time: fit with attribute probability set to True.

sklearn.svm.SVC — scikit-learn 0.22 documentation

得られた確率値を順位付けする場合、numpy配列のargsortメソッドを使えます。

>>> import numpy as np
>>> a = np.array([[0.2, 0.5, 0.12, 0.18], [0.1, 0.2, 0.5, 0.3]])
>>> a.argsort(axis=1)  # 昇順なので注意
array([[2, 3, 0, 1],
       [0, 1, 3, 2]])

Comments

[akibin-0219]

ありがとうございます！まさに求めていた事です、ありがとうございます！