1
我是TensorFlow的新手。我正在用我自己的數據集進行二進制分類。但是我不知道如何計算精度。任何人都可以幫助我做到這一點?如何計算TensorFlow中的CNN精度
我的分類器有5個卷積層,後面是2個完全連接的層。最後FC層具有2的輸出尺寸爲我所用:
prob = tf.nn.softmax(classification_features, name="output")
我是TensorFlow的新手。我正在用我自己的數據集進行二進制分類。但是我不知道如何計算精度。任何人都可以幫助我做到這一點?如何計算TensorFlow中的CNN精度
我的分類器有5個卷積層,後面是2個完全連接的層。最後FC層具有2的輸出尺寸爲我所用:
prob = tf.nn.softmax(classification_features, name="output")
纔算正確預測的百分比:
prediction = tf.argmax(prob, 1)
equality = tf.equal(prediction, correct_answer)
accuracy = tf.reduce_mean(tf.cast(equality, tf.float32))
能否請你在你的分類的樣子更具體? – Androbin
我的分類器有5個卷積層,其次是2個完全連接的層。最終FC層的輸出維數爲2,我已經使用了'prob = tf.nn.softmax(classification_features,name =「output」)' –
哦,真棒。謝謝 :) –