我是Tensorflow的新手。我在tensorflow会话和图形方面遇到问题。
我从训练应用程序中保存了权重,并且尝试在预测应用程序中根据测试集进行预测。
self.session = K.get_session()
with self.session.graph.as_default():
'''Define Model and load weights'''
self.model = nffm(linear_feature_columns,dnn_feature_columns,embedding_size=4,dnn_hidden_units=(4,4),task='binary')
self.model.load_weights(self.path_to_model_weight)
当我使用以下代码时,它会给出异常的预测。我使用了验证数据并尝试对其进行预测,但是并没有给出正确的预测。
with self.session.graph.as_default():
K.set_session(self.session)
prediction = self.model.predict(model_input)
但是当我在预测之前加载重量时,它就可以正常工作。
with self.session.graph.as_default():
K.set_session(self.session)
self.model.load_weights(self.path_to_model_weight)
prediction = self.model.predict(model_input)
由于我使用了相同的会话进行预测,因此我无法弄清楚为什么必须在预测之前加载权重?我认为这与张量流会话有关。