当前位置: 代码迷 >> 综合 >> 'ValueError: Tensor Tensor is not an element of this graph'(Keras, backend:TensorFlow)
  详细解决方案

'ValueError: Tensor Tensor is not an element of this graph'(Keras, backend:TensorFlow)

热度:139   发布时间:2023-09-24 04:34:44.0

在做在线负样本数据挖掘的时候,遇到了这个问题。

'ValueError: Tensor Tensor is not an element of this graph'

在线负样本数据挖掘涉及到在数据生成器中,使用模型对数据进行预测 ,也就是data generator()里面有model.predict()。

产生问题的原因:产生了两个计算图,TensorFlow不知道用哪一个。或者是load_model和model.predict()不在同一个线程。

参考网页:

https://zhuanlan.zhihu.com/p/27101000

https://github.com/keras-team/keras/issues/2397

解决方案为:

第一步:

import tensorflow as tf
graph = tf.get_default_graph()

第二步:

在数据生成器模型预测部分按照下面的方式写

with graph.as_default():prediction = model.predict()

 

 

  相关解决方案