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