当前位置:   article > 正文

使用MINST数据集训练模型并动态绘制Loss和准确率的趋势图_mnist数据集绘制训练损失和准确率曲线

mnist数据集绘制训练损失和准确率曲线

MINST训练的代码直接参考http://www.pianshen.com/article/3783475094/

由于显示的结果不太清晰明了,因此使用pyplot加工了下:代码如下

  1. #coding: utf-8
  2. import numpy as np
  3. import matplotlib.pyplot as plt
  4. import tensorflow as tf
  5. from tensorflow.examples.tutorials.mnist import input_data
  6. mnist = input_data.read_data_sets("MNIST_data", one_hot=True)
  7. input_x = tf.placeholder(tf.float32, [None, 28*28])/255
  8. output_y = tf.placeholder(tf.int32, [None, 10])
  9. input_x_images = tf.reshape(input_x, [-1, 28, 28, 1])
  10. test_x = mnist.test.images[:3000]
  11. test_y = mnist.test.labels[:3000]
  12. conv1 = tf.layers.conv2d(inputs=input_x_images,
  13. filters=32,
  14. kernel_size=[5, 5],
  15. strides=1,
  16. padding='same',
  17. activat
声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/我家小花儿/article/detail/354131
推荐阅读
相关标签
  

闽ICP备14008679号