卷积神经网络
这里使用了卷积神经网络,在上一个的基础上加入了卷积核,这样可以提到效果的准确性,但相应的加长了代码运行的时长,
运用3个卷积核可以将准确率提高到99.41%
代码实现
import tensorflow as tf
import random
from tensorflow.examples.tutorials.mnist import input_data
import matplotlib.pyplot as plt
tf.set_random_seed(1)
mnist = input_data.read_data_sets('MNIST_data/', one_hot=True)
lerning_rate = 0.001
training_times = 15
batch_size = 100
# 占位符
keep_prob = tf.placeholder(tf.float32)
X = tf.placeholder(tf.float32, [None, 784])
Y = tf.placeholder(tf.float32, [None, 10])
X_img = tf.reshape(X, [-1, 28, 28, 1])
# 初始化W(W为卷积核)
W1 = tf.Variable(tf.random_normal([3, 3, 1, 32], stddev=0.01))
W2 = tf.Variable(tf.random_normal([3, 3, 32, 64], stddev=0.01))
W3 = tf.Variable(tf.random_normal([3, 3, 64, 128], stddev=0.01))
# 卷积部分
L1 = tf.nn