笔记 - 模型训练:监控

打印 训练loss与测试loss

 print("Epoch", epoch, "MSE = ",
       sess.run(mse, feed_dict={
           X: X_train,
           y: y_train
       }))
 print("Epoch", epoch, "MSE = ",
       sess.run(mse, feed_dict={
           X: X_test,
           y: y_test
       }))

监控准确率

print("TrainSet batch acc : %s  " % accuracy.eval({x: batch_xs, y: batch_ys}))
print("ValidSet acc : %s" % accuracy.eval({x: my_mnist.validation.images, y: my_mnist.validation.labels}))

猜你喜欢

转载自blog.csdn.net/chen_holy/article/details/91413782