如何使用tensorflow库与sagemaker预处理

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处理器模块，可用处理模块的列表在以下文档中，可以在页面底部看到。

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

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