train_op = tf.train.AdamOptimizer(1e-3).minimize(loss)
loss_val, acc_val, _ = sess.run(
[loss, accuracy, train_op],
feed_dict={
x: batch_data,
y: batch_labels})
上面加了train_op,表示训练模型。
下面没有加train_op,表示用模型进行预测
test_acc_val = sess.run(
[accuracy],
feed_dict = {
x: test_batch_data,
y: test_batch_labels
})