auto ML 实验控制代码

虽然现在只要填几个参数，就可以让实验自动运行，但手动控制有时候还是会更机动一些，下面是主要代码，根据自己的情况填入各项参数：

from typing import Dict

from azureml.automl.core.shared.constants import ImageTask

from azureml.core import Experiment,workspace

from azure.train.automl import AutoMLImageConfig

from azureml.train.hyperdrive import GridParameterSamppling,choice

from torch.utils.data import Dataset

def main():

        ws=workspace.from_config()        #config.json:subscription_id,resource_group,workspace_name

        cluster_name="cluster_name"

        training_dataset_name="training_dataset_name"

        gridparameter_sampling={"model_name":choice("yolov5"),"validation_metric_type":"voc",}

        compute_target=ws.compute_targets[cluster_name]

        experiment=Experiment(ws,name="experiment_name")        #experiment name created on Azure

        config_yolov5=AutoMLImageConfig(

                task=ImageTask.IMAGE_OBJECT_DETECTION,

                compute_target=compute_target,

                training_data=training_dataset,

                validation_data=validation_dataset,

                hyperparameter_sampling=GripParameterSampling(gridparameter_sampling),

                iterations=1,

        )

        automl__run=experiment.submit(config_yolov5)

        automl_run.wait_for_completion(wait_post_processing=True)