基于Tensorflow实现MNIST softmax分类

import tensorflow as tf
from tensorflow.examples.tutorials.mnist import input_data
mnist=input_data.read_data_sets("MNIST_data/",one_hot=True)
x=tf.placeholder(tf.float32,[None,784])#x
y_=tf.placeholder(tf.float32,[None,10])#y

w=tf.Variable(tf.zeros([784,10]))#w
b=tf.Variable(tf.zeros([10])) #b
y=tf.nn.softmax(tf.matmul(x,w)+b) #y_

cross_entropy=tf.reduce_mean(-tf.reduce_sum(y_*tf.log(y)))#优化
train_step=tf.train.GradientDescentOptimizer(0.009).minimize(cross_entropy)

##会话
sess=tf.InteractiveSession()
tf.global_variables_initializer().run()

for i in range(1000):
    batch_x,batch_y=mnist.train.next_batch(100)
    sess.run(train_step,feed_dict={x:batch_x,y_:batch_y})

correct_pre=tf.equal(tf.arg_max(y,1),tf.arg_max(y_,1))
accuracy_rate=tf.reduce_mean(tf.cast(correct_pre,tf.float32))
print(sess.run(accuracy_rate,feed_dict={x:mnist.test.images,y_:mnist.test.labels}))

sess.close()