Python 如何将参数从SKLearn初始值设定项类传递到入口点（py.file）？

我正在尝试使用SKitLearn在Sagemaker建立一个培训工作。当我初始化SKLearn类时，我还希望添加一个可通过的参数，该参数将在train.py（入口点）中引用。我要传递的参数是一个模型类型字符串，如args.model\u type='classification'或args.model\u type='regression'

    sklearn_model = SKLearn(
        entry_point=train.py (I WANT TO PASS args.model_type into this PY.file, HOW??),
        role=role,
        train_instance_type=region_instance_dict[region],
        py_version="py3",
        sagemaker_session=sagemaker_session,
        train_max_run=MaxRuntimeInSeconds
        )

---

我将如何执行此操作？

您可以使用带有估计器构造函数的

超参数指定参数：

在您的情况下，hyperparameters dict中的项可能是
{--模型类型：}