web-dev-qa-db-ja.com

TensorFlowでCNNの精度を計算する方法

TensorFlowは初めてです。私は自分のデータセットで二項分類を行っています。ただし、精度の計算方法がわかりません。誰かがこれをするのを手伝ってくれませんか?

私の分類器には、5つの畳み込み層と、それに続く2つの完全に接続された層があります。最後のFCレイヤーの出力次元は2で、これを使用しました。

prob = tf.nn.softmax(classification_features, name="output")
7
akshata bhat

正しい予測のパーセンテージを計算するだけです。

prediction = tf.argmax(prob, 1)
equality = tf.equal(prediction, correct_answer)
accuracy = tf.reduce_mean(tf.cast(equality, tf.float32))
15
Androbin