予測結果の確率を取得したいです。

Question

前提・実現したいこと

https://teratail.com/questions/304425
「pytorchの_, preds = torch.max(outputs, 1)について」
の質問で、_は、その最大値(テンソル)を_で受けとっていますとのことですが、
要素の確率はどうやって取得すればよいのでしょうか？

要は、「アリ」である確率○○%と出力したいです。

試したこと

output_probability, preds = torch.max(outputs, 1)
としたのですが、確率ではなく、tensorが取得されました。

Answer 1

torch.max(outputs, 1)

outputs に、各出力(選択肢)毎の確からしさみたいな数値が入ってます

print(outputs)

ただし、outputs の全部の数値を足しても、1.0にはならないかもしれません
全部足して1.0になれば、outputs の各数値を100倍したら(%)として扱えますが、そうなるかどうかは、ネットワークの設計によります
ネットワークの最後にsoftmaxが入ってれば、全部足して1.0になります

softmaxの使い方は、たとえば下記を参考にしてください
PytorchでMNISTの分類をやってみる

Comments

[dthani]

返信おくれてすみません。softmaxで上手くいきました。ありがとうございます。