我可以将参数传递给SageMaker估计器的入口点吗？

Question

我正在使用SageMaker python sdk，希望传递一些参数供我的入口点使用，我不知道该怎么做。

from sagemaker.sklearn.estimator import SKLearn  # sagemaker python sdk

entrypoint = 'entrypoint_script.py'

sklearn = SKLearn(entry_point=entrypoint,  # <-- need to pass args to this
                  train_instance_type=instance_class,
                  role=role,
                  sagemaker_session=sm)

Answer 1

答案是否定的，因为Estimator基类或fit方法上没有任何参数可以接受传递给入口点的参数。

我通过将参数作为超参数字典的一部分传递来解决了这个问题。此get作为参数传递给入口点。

Answer 2

这里的entry_point参数是一个脚本文件，其中包含您要运行训练和预测的代码。

您可以参考example here

上面示例中使用的entry_point脚本为here