model_fn() 中 if mode == tf.estimator.ModeKeys.PREDICT: 不支持

Qihoo360 / hbox

AI on Hadoop

Apache License 2.0

1.73k stars 384 forks source link

model_fn() 中 if mode == tf.estimator.ModeKeys.PREDICT: 不支持 #40

Closed seeyourcell closed 6 years ago

seeyourcell commented 6 years ago

model = tf.estimator.Estimator(...) ps=model.predict(...)

xlearning好像直接跳过 ps=model.predict(...) 不执行，直接显示success

mode_fn() 中tf.estimator.ModeKeys.PREDICT 模块加print 函数，在日志中没有看到print 输出的内容用 model.eval(...) ,在model_fn(）中的 print内容能打印出来，从这些现象看 model.predict(...) 确实被忽略了

seeyourcell commented 6 years ago

解决了问题

Olivia-github commented 5 years ago

解决了问题

您好，我碰到了同样的问题，请问您是如何解决的呢？