在做在线负样本数据挖掘的时候,遇到了这个问题。
'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()