2016-04-06 5 views
1

ちょっと、私はCIFARの例で各クラスの確率を得る方法を理解しようとしています。現在、入力されたテンソルの予測クラスを入力して真偽を取り戻すことができますが、それが(10のうちの)クラスとその事実に当てはまる確率を教えてくれません。 sess.run()の後にこの情報を取得する方法はありますか?CIFARの例の出力の確率の取得

答えて

4

評価のためのCIFAR-10サンプルコード(cifar10_eval.py)は、各クラスに所与の画像がある確率を含むlogitsテンソルを(中間出力として)生成します。このテンソルをsess.run()に渡すと、画像の予測クラスがどのようなものかを示すために必要な情報が得られます。代わりにtf.nn.softmax(logits)を計算してsess.run()に渡すと、各クラスの確率(0〜1)が得られます。

関連する問題