aymericdamien преди 9 години
родител
ревизия
742675db9e
променени са 1 файла, в които са добавени 1 реда и са изтрити 1 реда
  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