import tensorflow as tf
from tensorflow.examples.tutorials.mnist import input_data
#参数设置
input_nodes = 784 #输入节点数
output_nodes = 10 #输出节点数
layer1_nodes = 500 #隐层节点数
bitch_size = 100 #每次训练包含的数据个数
learning_rate = 0.8 #初始学习率
learning_rate_deacy = 0.99 #学习率衰减率
l2_regulation = 0.0001 #l2正则化系数
moving_rate_deacy = 0.99 #滑动模型那个衰减率
train_num = 10000
#前向传播,variable_average平均滑动模型参数
def inference(x,variable_average,w1,b1,w2,b2):
if variable_average == None:
layer1 = tf.nn.relu(tf.matmul(x,w1)+b1)
return tf.matmul(layer1,w2)+b2
else:
layer1 = tf.nn.relu(tf.matmul(x,variable_average.average(w1))+variable_average.average(b1))
return tf.matmul(layer1,variable_average.average(w2))+variable_average.average(b2)
def train(mnist):
#feat