如何为Sagemaker打造火花训练图像

Question

如何在Amazon-sagemaker-examples / sagemaker-spark / pyspark_mnist / pyspark_mnist_custom_estimator.ipynb中将自定义火花算法的trainingImage作为kmeans trainingImage构建？

Answer 1

如果您想构建自己的算法以在SageMaker PySpark Estimator中使用，则可以按照以下示例来构建自己的算法容器：https://github.com/awslabs/amazon-sagemaker-examples/blob/master/advanced_functionality/scikit_bring_your_own/scikit_bring_your_own.ipynb

此示例逐步介绍了构建可用于训练和推理的自定义算法容器所需的步骤。