Могу ли я передать аргументы точке входа оценщика 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 или в методе подгонки нет параметра, который принимает аргументы для передачи в точку входа.

Я решил эту проблему, передав параметр в качестве части гиперпараметратолковый словарь.Это get передается точке входа в качестве аргументов.

Answer 2

Аргумент entry_point здесь для файла сценария, который содержит коды, которые вы хотите запустить для обучения и прогнозирования.

Вы можете обратиться к примеру здесь

А в приведенном выше примере сценария entry_point используется здесь