|
@@ -107,7 +107,7 @@ optimizer = tf.train.AdamOptimizer(learning_rate=learning_rate).minimize(cost)
|
|
|
|
|
|
# Evaluate model
|
|
|
correct_pred = tf.equal(tf.argmax(pred,1), tf.argmax(y,1))
|
|
|
-accuracy = tf.reduce_mean(tf.cast(correct_pred, tf.types.float32))
|
|
|
+accuracy = tf.reduce_mean(tf.cast(correct_pred, tf.float32))
|
|
|
|
|
|
# Initializing the variables
|
|
|
init = tf.initialize_all_variables()
|