TrainingOutput 클래스
파이프라인에서 사용할 특정 PipelineSteps의 특수 출력을 정의합니다.
TrainingOutput을 사용하면 자동화된 기계 학습 메트릭 또는 모델을 Azure Machine Learning 파이프라인의 다른 단계에서 사용할 단계 출력으로 사용할 수 있습니다. 와 함께 AutoMLStep 사용할 수 있습니다.HyperDriveStep
TrainingOutput을 초기화합니다.
param model_file: 출력에 포함할 특정 모델 파일입니다. 전용입니다 HyperDriveStep .
생성자
TrainingOutput(type, iteration=None, metric=None, model_file=None)
매개 변수
| Name | Description |
|---|---|
|
type
필수
|
학습 출력의 형식입니다. 가능한 값은 '메트릭', '모델'입니다. |
|
iteration
|
해당 학습 모델의 반복 번호입니다.
이 반복 번호는 'Model' 형식으로만 제공할 수 있습니다.
Default value: None
|
|
metric
|
최상의 학습 모델을 반환하는 데 사용할 메트릭입니다.
메트릭은 'Model' 형식으로만 제공할 수 있습니다.
Default value: None
|
|
model_file
|
출력에 포함할 특정 모델 파일입니다. 전용입니다 HyperDriveStep . Default value: None
|
|
type
필수
|
학습 출력의 형식입니다. 가능한 값은 '메트릭', '모델'입니다. |
|
iteration
필수
|
해당 학습 모델의 반복 번호입니다.
이 반복 번호는 'Model' 형식으로만 제공할 수 있습니다.
|
|
metric
필수
|
최상의 학습 모델을 반환하는 데 사용할 메트릭입니다.
메트릭은 'Model' 형식으로만 제공할 수 있습니다.
|
설명
TrainingOutput은 다른 단계가 생성한 메트릭 또는 모델을 PipelineDataPipeline사용할 수 있도록 하는 데 사용됩니다 AutoMLStepHyperDriveStep.
다음과 같이 AutoMLStep을 정의할 때 TrainingOutput을 사용합니다.
from azureml.pipeline.core import PipelineData, TrainingOutput
metrics_data = PipelineData(name='metrics_data', datastore=ds,
pipeline_output_name='metrics_output',
training_output=TrainingOutput(type='Metrics'))
model_data = PipelineData(name='model_data', datastore=ds,
pipeline_output_name='best_model_output',
training_output=TrainingOutput(type='Model'))
automl_step = AutoMLStep(name='automl_step',
automl_config=automl_config,
inputs=[input_data],
outputs=[metrics_data, model_data])
Notebook https://aka.ms/pl-automl에서 TrainingOutput 및 AutoMlStep 단계를 사용하는 예제를 참조하세요.