TrainingOutput クラス
パイプラインで使用するための特定の PipelineSteps の特殊な出力を定義します。
TrainingOutput を使用すると、自動化された機械学習メトリックまたはモデルを、Azure Machine Learning Pipeline の別のステップで使用されるステップ出力として使用できるようになります。 AutoMLStep または HyperDriveStep と同時に使用できます。
TrainingOutput を初期化します。
param model_file: 出力に含める特定のモデル ファイル。 HyperDriveStep の場合のみ。
- 継承
-
builtins.objectTrainingOutput
コンストラクター
TrainingOutput(type, iteration=None, metric=None, model_file=None)
パラメーター
名前 | 説明 |
---|---|
type
必須
|
トレーニング出力の種類。 指定できる値は、'Metrics'、'Model' です。 |
iteration
|
対応するトレーニング モデルのイテレーション番号。
このイテレーション番号は、種類が 'Model' の場合にのみ指定できます。
規定値: None
|
metric
|
最適なトレーニング モデルを返すために使用するメトリック。
メトリックは、種類が 'Model' の場合にのみ指定できます。
規定値: None
|
model_file
|
出力に含まれる特定のモデル ファイル。 HyperDriveStep の場合のみ。 規定値: None
|
type
必須
|
トレーニング出力の種類。 指定できる値は、'Metrics'、'Model' です。 |
iteration
必須
|
対応するトレーニング モデルのイテレーション番号。
このイテレーション番号は、種類が 'Model' の場合にのみ指定できます。
|
metric
必須
|
最適なトレーニング モデルを返すために使用するメトリック。
メトリックは、種類が 'Model' の場合にのみ指定できます。
|
注釈
TrainingOutput は、Pipeline を構築するときに PipelineData と共に使用して、他のステップで AutoMLStep または HyperDriveStep によって生成されたメトリックまたはモデルを使用できるようにします。
次のように 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])
ノートブック https://aka.ms/pl-automl で TrainingOutput と AutoMlStep のステップを使用する例を参照してください。