Tensorflow:张量流会话有问题

我是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)

由于我使用了相同的会话进行预测,因此我无法弄清楚为什么必须在预测之前加载权重?我认为这与张量流会话有关。

mengdegong 回答:Tensorflow:张量流会话有问题

暂时没有好的解决方案,如果你有好的解决方案,请发邮件至:iooj@foxmail.com
本文链接:https://www.f2er.com/3151530.html

大家都在问