如何在 sagemaker 预处理中使用 tensorflow 库

Question

我想在 sagemaker 管道中使用 TensorFlow 进行预处理。

但是，我还没有找到使用它的方法。

现在，我正在使用这个库进行预处理：

from sagemaker.sklearn.processing import SKLearnProcessor

framework_version = "0.23-1"

sklearn_processor = SKLearnProcessor(
    framework_version=framework_version,
    instance_type=processing_instance_type,
    instance_count=processing_instance_count,
    base_job_name="abcd",
    role=role,
)

现在，我需要使用 TensorFlow 进行预处理，但 python 模块无法导入 TensorFlow。

任何帮助将不胜感激。谢谢。

Answer 1

所以没有TensorFlow处理器模块，可用的处理模块列表在下面doc，可以在页面底部看到。

您可以使用 Processor 和 ScriptProcessor 类并将 TF 预处理作为脚本传递，其中包含必要模块的 requirements.txt。

我在 AWS 工作，我的意见是我自己的。