零基础学习tensorflow----模型的保存与载入（二）

没有meta这个文件，所以模型的结构要重新定义一次

import tensorflow as tf 
from tensorflow.examples.tutorials.mnist import input_data

mnist = input_data.read_data_sets("MNIST_data",one_hot = True)

batch_size = 64

n_batch = mnist.train.num_example

x = tf.placeholder(tf.float32,[None,784])
y = tf.placeholder(tf.float32,[None,10])

W = tf.Variable(tf.zeros([784,10])
b = tf.Variable(tf.zeros([10]))
prediction = tf.nn.softmax(tf.matmul(x,W)+b)

correct_prediction = tf.equal(tf.armax(y,1),tf.armax(prediction,1))
accuracy = tf.reduce_mean(tf.cast(correct_prediction,tf.flaot32))

saver = tf.train.Saver(max_to_keep=5)

loss = tf.losses.softmax_cross_entropy(y,prediction)
train_step = tf.train.AdamOptimizer(0.001).minimize(loss)