本文实例为大家分享了Tensorflow训练MNIST手写数字识别模型的具体代码,供大家参考,具体内容如下
import tensorflow as tf
from tensorflow.examples.tutorials.mnist import input_data
INPUT_NODE = 784 # 输入层节点=图片像素=28x28=784
OUTPUT_NODE = 10 # 输出层节点数=图片类别数目
LAYER1_NODE = 500 # 隐藏层节点数,只有一个隐藏层
BATCH_SIZE = 100 # 一个训练包中的数据个数,数字越小
# 越接
1