Tensorflow学习:手写数字识别完整版(准确率高达98%)

Tensorflow学习:手写数字识别完整版(准确率高达99.9%)

# 解决过拟合的方法:dropout
import tensorflow as tf
from tensorflow.examples.tutorials.mnist import input_data

# 载入数据集
mnist = input_data.read_data_sets("MNST_data", one_hot=True)

# 每个批次的大小:一次性向神经网络中放入100张图片进行训练:以矩阵的形式放进去
batch_size = 100
# 计算一共多少个批次 //:整除
n_batch = mnist.train.num_examples // batch_size

# 定义两个placeholder  None:过会以一个批次喂进去,784:28*28(把一个图片拉成一个784的向量) 10:输出时0-9的10个数
x = tf.placeholder(tf.float32, [None, 784])
y = tf.placeholder(tf.float32, [None, 10])
# 定义一个placeholder:设置dropout的参数
keep_prob = tf.placeholder(tf.float32)
#定义一个学习率的变量
lr = tf.Variable(0.001, dtype = tf.float32)

# 定义一个简单的神经网络(输入层784个神经元,输出层10个神经元,增加隐层)
# 定义权值:初始化为0并不是一种很好地初始化的方式
# W = tf.Variable(tf.zeros([784, 10]))
# 给权值初始化的时候:使用截断的产生正态分布的函数,标准差为0.1
W1 = tf.Variable(tf.truncated_normal([784, 500], stddev=0.1))
# 定义偏置值:初始化为0并不是一种很好地初始化的方式
# b = tf.Variable(tf.zeros([10]))
b1 = tf.Variable(tf.zeros([500]) + 0.1)
# 定义L1的输出
L1 = tf.nn.tanh(tf.matmul(x, W1) + b1)
# 定义L1的dropout:keep_prob:可以设置有百分之多少的神经元是工作的,等于1时是百分之百的神经元在工作
L1_drop = tf.nn.dropout(L1, keep_prob)

'''增加2个隐藏层:第一个隐藏层有500个神经元,第二个隐藏层有300个神经元'''
W2 = tf.Variable(tf.truncated_normal([500, 300]
  • 1
    点赞
  • 19
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
Keras是一种用户友好的深度学习库,能够简化卷积神经网络(CNN)的实现过程。通过使用Keras库,我们可以非常轻松地实现手写数字识别的模型,并达到99.6%的准确率。 首先,我们需要准备一个适用于手写数字识别的数据集,如MNIST数据集。MNIST数据集包含了大量的手写数字图像,每个图像都有对应的标签(数字)。我们可以使用Keras内置的函数来加载和预处理这个数据集,使其适用于训练模型。 接下来,我们可以定义一个CNN模型。在Keras中,我们可以使用Sequential模型来构建我们的网络。通过添加不同的层,我们可以定义卷积层、池化层和全连接层。这些层可以通过简单的方法来添加到我们的模型中。 然后,我们可以使用compile()方法来编译我们的模型。我们需要指定损失函数、优化器和评估指标。在这种情况下,我们可以选择交叉熵作为损失函数,Adam优化器作为优化器,并选择准确率作为评估指标。 接下来,我们需要使用fit()方法来训练我们的模型。我们可以指定训练数据集、训练的轮数和每批次的大小。Keras会自动计算并更新权重和偏差来最小化损失函数。 最后,我们可以使用evaluate()方法来评估我们的模型在测试数据集上的表现。这将给出模型的准确率,即它正确分类的图像所占的比例。 通过以上步骤,我们可以使用Keras实现一个CNN模型来进行手写数字识别,且其准确率可以达到99.6%。这是一个非常令人满意的结果,表明我们的模型在识别手写数字方面具有很高的准确性。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值