首页 > 解决方案 > Tornado API 2.2.5 和 2.3.1 之间的 Keras 预测?

问题描述

我在 keras 版本 2.2.5 和 python 3.7.4 中的 tensorflow 1.14.0 中构建了 keras 顺序模型。为了在 Tornado API 中进行预测,我使用如下图初始化,它工作得非常好

global graph
graph = tf.get_default_graph()        
with graph.as_default():
    prediction = model.predict_proba(X)

如果我升级 keras == 2.3.1,我会收到以下错误

tensorflow.python.framework.errors_impl.InvalidArgumentError: Tensor dense_1_input:0, specified in either feed_devices or fetch_devices was not found in the Graph

我可以知道如何在新版本中修复

标签: pythontensorflowkerastornado

解决方案


推荐阅读