在线时间:8:00-16:00
迪恩网络APP
随时随地掌握行业动态
扫描二维码
关注迪恩网络微信公众号
30行代码奉上!(MNIST手写数字的识别,识别率大约在91%,简单尝试的一个程序,小玩具而已) 1 import tensorflow.examples.tutorials.mnist.input_data as input_data 2 import tensorflow as tf 3 mnist = input_data.read_data_sets('/temp/', one_hot=True) 4 5 #设置 6 x = tf.placeholder(tf.float32,[None,784]) 7 W = tf.Variable(tf.zeros([784,10])) 8 b = tf.Variable(tf.zeros([10])) 9 #预测值 10 y = tf.nn.softmax(tf.matmul(x,W)+b) 11 #真值 12 y_ = tf.placeholder(tf.float32,[None,10]) 13 #交叉熵 14 cross_entropy = -tf.reduce_sum(y_*tf.log(y)) 15 16 #使用优化器 17 train_step = tf.train.GradientDescentOptimizer(0.01).minimize(cross_entropy) 18 #初始化变量 19 init = tf.initialize_all_variables() 20 #创建对话 21 sess = tf.Session() 22 sess.run(init) 23 24 25 for i in range(1000): 26 batch_xs,batch_ys = mnist.train.next_batch(100) 27 sess.run(train_step,feed_dict={x:batch_xs,y_:batch_ys}) 28 29 correct_prediction = tf.equal(tf.argmax(y,1), tf.argmax(y_,1)) 30 accuracy = tf.reduce_mean(tf.cast(correct_prediction, "float")) 31 print(sess.run(accuracy, feed_dict={x: mnist.test.images, y_: mnist.test.labels}))
1 import tensorflow.examples.tutorials.mnist.input_data as input_data 2 import tensorflow as tf 3 mnist = input_data.read_data_sets('/temp/', one_hot=True) #设置 6 x = tf.placeholder(tf.float32,[None,784]) 7 W = tf.Variable(tf.zeros([784,10])) 8 b = tf.Variable(tf.zeros([10])) 9 #预测值 10 y = tf.nn.softmax(tf.matmul(x,W)+b) 11 #真值 12 y_ = tf.placeholder(tf.float32,[None,10])
其中x作为输入是一个1x768的向量,然后就是经过权重和偏食,就得到10个输出,然后用softmax()进行预测值的输出。 此外y_作为真值,要用到一个占位符。
主要用到的tensorflow的函数有 tf.placeholder 设置一个占位符,用于设置输入
tf.Variable 设置一个变量,用于设置权重和偏置
tf.nn.softmax softmax回归函数
tf.matmul 实现矩阵相乘
13 #交叉熵
14 cross_entropy = -tf.reduce_sum(y_*tf.log(y))
15
16 #使用优化器
17 train_step = tf.train.GradientDescentOptimizer(0.01).minimize(cross_entropy)
18 #初始化变量
19 init = tf.initialize_all_variables()
20 #创建对话
21 sess = tf.Session()
22 sess.run(init)
使用交叉熵作为误差评判标准,并以此来实现随机梯度下降。交叉熵函数公式如下:
tf.reduce_sum 求和
tf.log 求对数
tf.train.GradientDescentOptimizer(learnrate) 创建一个优化器对象,使用的是随机梯度下降的方法,参数可以设置学习率
tf.train.GradientDescentOptimizer对象的minimize()方法 指定损失函数才能进行优化
tensorflow函数:
tf.initialize_all_variables() 创建一个初始化所有变量的对象
tf.Session() 创建一个Session对话
sess.run() Session对话指定执行操作
25 for i in range(1000): 26 batch_xs,batch_ys = mnist.train.next_batch(100) 27 sess.run(train_step,feed_dict={x:batch_xs,y_:batch_ys}) 28 29 correct_prediction = tf.equal(tf.argmax(y,1), tf.argmax(y_,1)) 30 accuracy = tf.reduce_mean(tf.cast(correct_prediction, "float")) 31 print(sess.run(accuracy, feed_dict={x: mnist.test.images, y_: mnist.test.labels})) mnist.train.next_batch 可以返回一个小批数据集 tf.argmax 返回的是vector中的最大值的索引号
tf.equal 判断是否相等
tf.reduce_mean 求均值
tf.cast 类型转换
|
请发表评论