导入tf的问题转换为云ml

时间:2018-02-20 18:22:32

标签: python tensorflow google-cloud-ml tensorflow-transform

每当我尝试在mlengine作业上导入tensorflow-transform时,我都会遇到以下问题:

Traceback (most recent call last):
File "/usr/lib/python2.7/runpy.py", line 162, in _run_module_as_main
"main", fname, loader, pkg_name)
File "/usr/lib/python2.7/runpy.py", line 72, in _run_code
exec code in run_globals
File "/root/.local/lib/python2.7/site-packages/trainer/task.py", line 
15, in 
from tensorflow.contrib.boosted_trees.python.ops import quantile_ops
ImportError: No module named boosted_trees.python.ops

Tensorflow转换在数据流上运行良好,但是当我尝试训练模型时,会出现上述错误。 Tensorflow一般似乎在mlengine上工作正常,但如果我尝试只导入boosted_trees.python.ops,我会遇到问题。我使用的是tf 1.4和tft 0.4.0。我正在运行的代码是cloudml-samples reddit_tft示例的略微修改版本。

1 个答案:

答案 0 :(得分:0)

AFAI python包是'tensorflow_transform'而不是'tensorflow-transform'。 您的代码中的'tensorflow'和'tensorflow_transform'之间存在冲突。我在reddit_tft示例中使用了tensorflow_transform版本0.5和tensorflow 1.4.1并且它工作正常。