提问人:Egorsky 提问时间:11/10/2023 最后编辑:wade zhou - MSFTEgorsky 更新时间:11/13/2023 访问量:70
如何在 argparse 中为 AzureML 中的管道添加元组?
How to add tuple in argparse for the pipeline in AzureML?
问:
我想为我在管道中执行的函数进行元组。为简单起见,我将跳过读取数据和其他与主题无关的步骤。这看起来像这样: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>
string
integer
number
bool
有没有办法解析其中的元组?或者唯一的方法是解析和单独并将它们组合成 main 函数中的元组?p, d, q
P, D, Q, S
答: 暂无答案
评论