2017-03-05 436 views
1

我是TensorFlow的新手。我正在用我自己的數據集進行二進制分類。但是我不知道如何計算精度。任何人都可以幫助我做到這一點?如何計算TensorFlow中的CNN精度

我的分類器有5個卷積層,後面是2個完全連接的層。最後FC層具有2的輸出尺寸爲我所用:

prob = tf.nn.softmax(classification_features, name="output") 
+0

能否請你在你的分類的樣子更具體? – Androbin

+1

我的分類器有5個卷積層,其次是2個完全連接的層。最終FC層的輸出維數爲2,我已經使用了'prob = tf.nn.softmax(classification_features,name =「output」)' –

+0

哦,真棒。謝謝 :) –

回答

4

纔算正確預測的百分比:

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