TensorFlow中sess.run()
在训练和预测时传入的参数有所不同。在训练时,通常需要传入训练数据、目标值和训练模型所需要的参数,如优化器、学习率等。在预测时,通常只需要传入测试数据和预测模型所需要的参数。
示例代码:
# 训练
with tf.Session() as sess:
sess.run(tf.global_variables_initializer())
for i in range(num_steps):
# 获取训练数据和目标值
x_batch, y_batch = get_next_batch()