如何在 argparse 中为 AzureML 中的管道添加元组?

How to add tuple in argparse for the pipeline in AzureML?

提问人:Egorsky 提问时间:11/10/2023 最后编辑:wade zhou - MSFTEgorsky 更新时间:11/13/2023 访问量:70

问:

我想为我在管道中执行的函数进行元组。为简单起见,我将跳过读取数据和其他与主题无关的步骤。这看起来像这样:argparse

def model_train_sales(X_train, order: tuple, seasonal_order: tuple):

    model = sm.tsa.SARIMAX(X_train['sales'], order=order, seasonal_order=seasonal_order)
    results = model.fit()

    return model, results

def main():

    parser = argparse.ArgumentParser()

    parser.add_argument("--order", type=tuple)
    parser.add_argument("--seasonal_order", type=tuple)

    args = parser.parse_args()

    model, results = model_train_sales(X_train['sales'], order=args.order, 
    seasonal_order=args.seasonal_order)

目前一切正常,但是当您开始构建管道时,解析变量的类型是不同的。

from azure.ai.ml import command
from azure.ai.ml import Input, Output

demo_model_training_component = command(
    name='my sarima pipeline',
    display_name='my description',
    description='A long description.',
    inputs={
        "order": Input(type='<TYPE>'),
        "seasonal_order": Input(type='<TYPE>'),
    },
    outputs=dict(
        df = Output(type="uri_folder", mode="rw_mount")
    ),
    code = feature_creation_src_dir,
    command = """python sarima_model.py \
              --order ${{inputs.order}} --seasonal_order ${{inputs.seasonal_order}} \
              --df ${{outputs.df}}
              """,
    environment = f"{pipeline_job_env.name}"{pipeline_job_env.version}",
)

在这里,我签署了我不确定应该哪种类型的地方。我知道类型是有限的,要么是 ,要么是 。<TYPE>stringintegernumberbool

有没有办法解析其中的元组?或者唯一的方法是解析和单独并将它们组合成 main 函数中的元组?p, d, qP, D, Q, S

python-3.x 机器学习 元组 azure-machine-learning-service

评论


答: 暂无答案