我已经在一些数据上训练了一个模型(只是一个简单的分类任务)。之后,我希望使用相同的模型通过单独的函数make_prediction()
运行一些预测。
所以目前我的主文件只是:
agent.train(data)
agent.make_predictions(new_data)
和tf.global_variables_initializer()
with tf.Session() as sess:
sess.run(init)
我没有在第二个函数中初始化变量,因此会话与之前的会话不同,但令我惊讶的是,我无法重新打开之前的会话。我是否需要在训练后检查模型,然后每次重新加载?
非常感谢