TensorFlow深度学习2--手写数学图片识别初体验

1、 手写数字图片数据集
在这里插入图片描述
手写数字图片数据集,它包含了0~9共10种数字的手写图片,采集自不同书写风格的真实手写图片,一共70000张图片。其中60000张图片作为训练集,用来训练模型,剩下10000张图片作为测试集,用来预测或者测试,训练集和测试集共同组成了整个MNIST数据集。
2、 模型搭建

(x, y), (x_val, y_val) = datasets.mnist.load_data()  #加载MNIST数据集
x = tf.convert_to_tensor(x, dtype=tf.float32) / 255.    #转换为浮点数
y = tf.convert_to_tensor(y, dtype=tf.int32)            #转换为整形张量
y = tf.one_hot(y, depth=10)                  #one-hot 编码
print(x.shape, y.shape)
train_dataset = tf.data.Dataset.from_tensor_slices((x, y))     #构建数据集
train_dataset = train_dataset.batch(200)                       #批量训练
#利用Sequential容器封装3个网络层,前网络层的输出默认作为下一层的输入
model = keras.Sequential([ 
    layers.Dense(512, activation='relu'),
    layers.Dense(256, activation='relu'),
    layers.Dense(10)])
optimizer = optimizers.SGD(learning_rate=0.001)

3、 模型训练

def train_epoch(epoch):
    losses=[]
    # Step4.loop
    for step, (x, y) in enumerate(train_dataset):
        with tf.GradientTape() as tape:    #构建梯度记录环境
            # 打平操作:[b, 28, 28] => [b, 784]
            x = tf.reshape(x, (-1, 28*28))
            # Step1. compute output
            # [b, 784] => [b, 10]
            out = model(x)
            # Step2. compute loss
            loss = tf.reduce_sum(tf.square(out - y)) / x.shape[0]
        # Step3. 自动计算参数的梯度  w1, w2, w3, b1, b2, b3
        grads = tape.gradient(loss, model.trainable_variables)
        # w' = w - lr * grad  更新网络参数
        optimizer.apply_gradients(zip(grads, model.trainable_variables))
        if step % 100 == 0:
            print(epoch, step, 'loss:', loss.numpy())
    return loss

4、 模型运行

def train():
    losses = []
    for epoch in range(50):
        loss=train_epoch(epoch)
        losses.append(float(loss))
    return  losses
if __name__ == '__main__':
    losses =train()
    plt.figure()
    x = [i  for i in range(len(losses))]
    plt.plot(x, losses, color='C0', marker='s', label='训练')
    plt.ylabel('MSE')
    plt.xlabel('Epoch')
    plt.legend()
    plt.savefig('train.png')
    plt.show()

5、 MNIST数据集的训练误差曲线
在这里插入图片描述

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值