問題タブ [argmax]

私は単純ベイズ分類器に取り組んでおり、ログ予測として次の配列のリストを取得しました。

np.argmax(predict, axis = 1)の最大値を選択するために、実行しようとしていますpredict。対応するインデックスを返すために使用argmaxしましたが、このエラーをスローし続けます: TypeError: 'list' object is not callable. なぜそのエラーがスローされるのですか?

セグメンテーションの件名で異なるバッチのプロット出力について疑問と質問があります。

以下のスニペットは、各クラスの確率と予測出力をプロットしています。

prob plots が 1 つのバッチをプロットしていることは確かですが、torch.argmax(outputs, 1) を取得したときの予測についてはわかりません。ネットワークの出力のサイズが [10,4,256,256] のときに、1 つのバッチの argmax をプロットしましたか。

また、バッチ サイズが 10 の場合にすべてのバッチの予測をプロットするにはどうすればよいでしょうか。