ソフトマックス層で2番目に大きな確率を取るラベルを確認する方法はありますか?
1 次查看(过去 30 天)
显示 更早的评论
分類用の畳み込みニューラルネットワークに関する質問になります。
入力データに対してソフトマックス層では最も大きな確率をとるラベルを出力するようになっていると思うのですが、
2番目に大きな確率をとるラベルを確認する方法はありますでしょうか。
(例)数字8のイメージデータに対して、ソフトマックス層でラベル「8」を出力した時に2番目に大きな確率をとるラベル「9」を確認する。
ご教示ください。
よろしくお願いいたします。
0 个评论
采纳的回答
Kazuya
2019-10-2
ラベルの「確認」がどんな処理か分かりませんが、泥臭い方法でよければ思いつきます。
まず提示いただいた例題ですと
最後の
YPred = classify(net,imdsValidation);
の代わりに
scores = activations(net,imdsValidation,'softmax','OutputAs','columns');
と実行すれば各ラベル毎の確率にあたるモノを出力できます。
ここから例えば
[sortedScore,I] = sort(scores,'descend');
とソートします。
第二出力 I を追うことで、2番目に数値が高かったラベルを認識できます。
例えば
labels = net.Layers(15).Classes;
firstPred = labels(I(1,:)); % 1番目ラベル
secondPred = labels(I(2,:)); % 2番目のラベル
といった具合です。
isequal(firstPred,YPred)
とすれば
YPred = classify(net,imdsValidation);
で計算された YPred と firstPred が同値であることが確認できます。
更多回答(0 个)
另请参阅
类别
在 Help Center 和 File Exchange 中查找有关 イメージを使用した深層学習 的更多信息
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!