2017-03-05 8 views
1

Ich bin neu in TensorFlow. Ich mache eine binäre Klassifizierung mit meinem eigenen Datensatz. Ich weiß jedoch nicht, wie ich die Genauigkeit berechnen soll. Kann mir bitte jemand dabei helfen?So berechnen Sie die Genauigkeit von CNN in TensorFlow

Mein Klassifikator hat 5 Faltungsschichten, gefolgt von 2 vollständig verbundenen Schichten. Die endgültige FC Schicht haben eine Ausgangs Dimension von 2, für die ich verwendet:

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

Könnten Sie bitte etwas konkreter, wie Sie Ihre Klassifikator aussieht? – Androbin

+1

Mein Klassifikator hat 5 Faltungsschichten gefolgt von 2 vollständig verbundenen Schichten. Die letzte FC-Ebene hat eine Ausgabe-Dimension von 2, für die ich 'prob = tf.nn.softmax (classification_features, name =" output ")' –

+0

Oh genial verwendet habe. Vielen Dank :) –

Antwort

4

berechnen Sie einfach den Prozentsatz der richtigen Vorhersagen:

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