Tensorflow：予測値を得る

Question

私はMLPのサンプルコードをダウンロードし、その動作を理解しようとしています。私はテストデータセットの精度を計算するこのラインを越えて来ました。

accuracy = sess.run(accuracy, feed_dict={X: data_test, y: labels_test, dropout_keep_prob:1.}) 

ここでも、予測ラベルを取得したいと考えています。予測ラベルを取得するにはどうすればよいですか？

Answer 1

予測テンソルを取得する必要があります。コードをお持ちの場合は、それをyと比較して精度を計算します。 predictionと言うと、次のように書くことができます。

accuracy, prediction = sess.run([accuracy, prediction], feed_dict={X: data_set, y:labels_test, ...})