We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 8dc3c04 commit 86a663aCopy full SHA for 86a663a
README.md
@@ -83,7 +83,7 @@ network = tl.layers.DenseLayer(network, n_units=10, act = tf.identity, name='out
83
84
# Define cost function and metric.
85
y = network.outputs
86
-cost = tl.cost.cross_entropy(y, y_)
+cost = tl.cost.cross_entropy(y, y_, 'cost')
87
correct_prediction = tf.equal(tf.argmax(y, 1), y_)
88
acc = tf.reduce_mean(tf.cast(correct_prediction, tf.float32))
89
y_op = tf.argmax(tf.nn.softmax(y), 1)
0 commit comments