1

每当我尝试在 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 示例的略微修改版本。

4

1 回答 1

0

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

于 2018-03-03T04:37:45.670 回答