python 张量流均值平方误差MSE

Posted 2021-05-09

# create Y_predictions, in this case linear regression
Y_predictions = tf.add(b, tf.multiply(w,X))

squares = tf.square(Y - Y_predictions)

loss = tf.reduce_mean(squares, name='loss')

#gradiaent descent. pass learning rate as .01
optimizer = tf.train.GradientDescentOptimizer(0.01).minimize(loss=loss)