MNIST-tensorboard-网页可视化查看结果

本文档介绍了如何配置环境并使用TensorFlow搭建5层神经网络训练MNIST数据集。步骤包括数据预处理、模型构建、学习率设置、梯度下降以及日志保存。通过在命令行运行特定命令后,可以利用TensorBoard在本地浏览器中以6006端口查看训练过程和结果,实现训练过程的可视化。同时,文章也提醒遇到问题要学会借助搜索引擎解决。
摘要由CSDN通过智能技术生成

“历史遗留”的学习记录


MNIST手写数字—tensorflow2—tensorboard—网页可视化查看结果
(PS:时隔久远,但还是能运行)

  1. 子程序包括:数据集处理、数据集图片读取转换、PNG图显示
  2. 对数据训练集、验证集的打乱
  3. 搭建5层神经网络
  4. 设置学习率并进行梯度下降
  5. 保存运行日志在文件夹中
  6. 用验证(测试)集评估准确率
  7. 在网页上显示过程并展示部分图片
import  tensorflow as tf
from    tensorflow.keras import datasets, layers, optimizers, Sequential, metrics
import  datetime
from    matplotlib import pyplot as plt
import  io

# 对数据集预处理,转换数据类型
def preprocess(x, y):
    x = tf.cast(x, dtype=tf.float32) / 255.
    y = tf.cast(y, dtype=tf.int32)
    return x,y

# 将数据集的图片读取、转换成PNG格式并返回
def plot_to_image(figure):
  buf = io.BytesIO()     #在内存中读写bytes
  plt.savefig(buf, format='png')       #保存生成的图片
  plt.close(figure)
  buf.seek(0)       #文件读取指针移动到文件第“0”个字节处,这里即移动到文件头位置
  image = tf.image.decode_png(buf.getvalue(), channels=4)     #对PNG图像进行解码,得到图像的像素值,用于显示图像
  image = tf.expand_dims(image, 0)    #指定在第“0”维添加维数
  return image

# 将上面返回的PNG图像显示出来
def image_grid(images):
  figure = plt.figure(figsize=(10,10))   #创建宽和高分别为10英寸的图像
  for i in range(64):
    plt.subplot(8, 8, i + 1, title='name')  #生成36个子图像
    plt.xticks([])
    plt.yticks([])
    plt.grid(False)
    plt.imshow
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值