Tensorflow入门教程-1

这篇TensorFlow入门教程介绍了MNIST手写数字数据集,包括数据的导入和描述,并讲解了Softmax回归的概念及神经网络结构。通过Python代码展示了如何实现模型,包括占位符、矩阵相乘和学习率等关键概念。
摘要由CSDN通过智能技术生成

详细参见我的独立博客 http://shuaizhang.tech/2016/11/29/Tensorflow%E6%95%99%E7%A8%8B1-MNIST/

该教程大部分的材料来自tensorflow官网tensorflow.

MNIST数据描述

MNIST数据是一个非常经典的机器学习的数据集,它是一个图片数据集,每张有一个手写的阿拉伯数字从0到9,该数据集地址为MNIST, 在Python当中,我们可以使用代码来导入数据,代码如下:
{% highlight python %}
from tensorflow.examples.tutorials.mnist import input_data
mnist = input_data.read_data_sets(“MNIST_data/”, one_hot=True)
{% endhighlight %}

MNIST数据有三个部分,训练数据(55000条 mnist.train), 测试数据(10000条 mnist.test)以及验证数据(5000条 mnist.validation),这个样的数量分割其实是非常重要的。在做模型训练和测试的时候,一般会将数据按照一定的比例(比如8-2法则)分割。
MNIST的每张图片有28×28个像素,相对真实的图片数据这个已经很简化了,看图片示例:
1

mnist.train.images 类型为tensor数据维度为[55000,28*28]; mnist.train.labels的维度为[55000,10],这里的10意思:假如这个数字代表9,则为[0,0,0,0,0,0,0,0,0,1]

Softmax回归

如果不了解softmax,可以先回顾一下sigmoid,sigmoid用在分类结果只有2个的情况下,softmax用在分类结果有很多种的情况下,详细参见softmax,下图给出了整个神经网络的结构:
2

其中W代表权重,b代表bias,softmax对计算出来的结果再进行一次处理,将其映射到0到9不同的标签上面去

3

计算公式为:
\(y=softmax(Wx+b)\)

代码实现

{% highlight python %}
from tensorflow.examples.tutorials.mnist import input_data
import tensorflow as tf
import argparse
import sys

FLAGS = None

def main(_):
mnist = input_data.read_data_sets(FLAGS.data_dir,one_hot=True)

x = tf.placeholder(tf.float32,[None, 784])
W = tf.Variable(tf.zeros([784,10]))
b = tf.Variable(tf.zeros([10]))
y = tf.matmul(x,W) + b #matmul means matrix multiplication

#defince the loss and optimizer
y_ = tf.placeholder(tf.float32,[None,10])

cross_entropy = tf.reduce_mean(tf.nn.softmax_cross_entropy_with_logits(y,y_))
train_step = tf.train.GradientDescentOptimizer(0.5).minimize(cross_entropy)#o.5  is learning rate

sess = tf.InteractiveSession()

#train
tf.global_variables_initializer().run()

for _ in range(1000): # we will run the train step 1000 times
    batch_xs, batch_ys = mnist.train.next_batch(100) # batch size, for each iteration, we only use 100 points of the dataset
    sess.run(train_step, feed_dict={x: batch_xs, y_: batch_ys})


#test
correct_prediction = tf.equal(tf.argmax(y,1), tf.argmax(y_,1))
accuracy = tf.reduce_mean(tf.cast(correct_prediction, tf.float32))
print(sess.run(accuracy, feed_dict={x: mnist.test.images, y_: mnist.test.labels}))

if name == ‘main‘:
parser = argparse.ArgumentParser()
parser.add_argument(‘–data_dir’, type=str, default=’dataset’, help=”Directory for storing data”)
FLAGS, unparsed = parser.parse_known_args()
tf.app.run(main = main, argv=[sys.argv[0]] + unparsed)

{% endhighlight %}

下面,我们对上面代码中存在疑问的地方进行解释

  • placeholder

这个我们可以理解为占位符,后面的feed_dict中我们要给他具体的值,否则会报错,例如上面代码中我们有x和\(y\_\)两个placeholder,因此,需要在feed_dict中给出他们的具体值

  • matmul

这个代表矩阵相乘,因为tensorflow需要适应GPU的运算,所以我们不能再使用python的Numpy的适合CPU的运算方法,因为tensorflow定义了一套自己的运算

  • Learning rate

代码中的0.5, learning rate就是梯度下降算法中的,每一步向下走的距离,可以参考梯度下降算法

  • batch size

代码中的100, batch size是stochastic gradient descent-随机梯度下降算法中特有的,传统的梯度下降一次需要训练非常多的数据,所以运算复杂度非常高,因此我们在训练的时候每次都只是取其中的一部分来进行训练,batch size就是取出来的数据量的大小。想想我们总共只有55000条数据,但是我们循环了1000次,每次100条,总共有100000条数据了,所以其中肯定是有重复的训练的数据。

我的最后的训练结果为0.91918,每次训练都不一样,将参数换一下训练的结果也会不一样,有兴趣的可以自己调试一下!

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值