Skip to content

Instantly share code, notes, and snippets.

@IntegerMan
Created July 19, 2022 03:16
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save IntegerMan/5dde01cf57f8c05bfcfbe9da4d55aa9e to your computer and use it in GitHub Desktop.
Save IntegerMan/5dde01cf57f8c05bfcfbe9da4d55aa9e to your computer and use it in GitHub Desktop.
# Create the configuration for the experiment
from azureml.train.automl import AutoMLConfig
# See https://docs.microsoft.com/en-us/python/api/azureml-train-automl-client/azureml.train.automl.automlconfig.automlconfig?view=azure-ml-py for details
automl_config = AutoMLConfig(
task='regression', # regression, classification, or forecasting
training_data=ds, # The data to use to train the model
label_column_name='thingIWantToPredict', # The column we're trying to predict
n_cross_validations=3, # How many cross-validation sets to use
primary_metric='normalized_root_mean_squared_error',# The metric we use to compare model performance
compute_target=cpu_cluster, # Where the experiment should be run
max_concurrent_iterations=max_nodes, # How many models can be trained simultaneously
iterations=40, # The total number of models to train
iteration_timeout_minutes=5 # The amount of time before giving up on a single model training run
)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment