aymericdamien %!s(int64=9) %!d(string=hai) anos
pai
achega
742675db9e
Modificáronse 1 ficheiros con 1 adicións e 1 borrados
  1. 1 1
      examples/2 - Basic Classifiers/logistic_regression.py

+ 1 - 1
examples/2 - Basic Classifiers/logistic_regression.py

@@ -32,7 +32,7 @@ b = tf.Variable(tf.zeros([10]))
 activation = tf.nn.softmax(tf.matmul(x, W) + b) # Softmax
 
 # Minimize error using cross entropy
-cost = -tf.reduce_sum(y*tf.log(activation)) # Cross entropy
+cost = tf.reduce_mean(-tf.reduce_sum(y*tf.log(activation), reduction_indices=1)) # Cross entropy
 optimizer = tf.train.GradientDescentOptimizer(learning_rate).minimize(cost) # Gradient Descent
 
 # Initializing the variables