Train Model module

This article describes a module in Azure Machine Learning designer.

Use this module to train a classification or regression model. Training takes place after you have defined a model and set its parameters, and requires tagged data. You can also use Train Model to retrain an existing model with new data.

How the training process works

In Azure Machine Learning, creating and using a machine learning model is typically a three-step process.

  1. You configure a model, by choosing a particular type of algorithm, and defining its parameters or hyperparameters. Choose any of the following model types:

    • Classification models, based on neural networks, decision trees, and decision forests, and other algorithms.
    • Regression models, which can include standard linear regression, or which use other algorithms, including neural networks and Bayesian regression.
  2. Provide a dataset that is labeled, and has data compatible with the algorithm. Connect both the data and the model to Train Model.

    What training produces is a specific binary format, the iLearner, that encapsulates the statistical patterns learned from the data. You cannot directly modify or read this format; however, other modules can use this trained model.

    You can also view properties of the model. For more information, see the Results section.

  3. After training is completed, use the trained model with one of the scoring modules, to make predictions on new data.

How to use Train Model

  1. Add the Train Model module to the pipeline. You can find this module under the Machine Learning category. Expand Train, and then drag the Train Model module into your pipeline.

  2. On the left input, attach the untrained mode. Attach the training dataset to the right-hand input of Train Model.

    The training dataset must contain a label column. Any rows without labels are ignored.

  3. For Label column, click Edit column in the right panel of module, and choose a single column that contains outcomes the model can use for training.

    • For classification problems, the label column must contain either categorical values or discrete values. Some examples might be a yes/no rating, a disease classification code or name, or an income group. If you pick a noncategorical column, the module will return an error during training.

    • For regression problems, the label column must contain numeric data that represents the response variable. Ideally the numeric data represents a continuous scale.

    Examples might be a credit risk score, the projected time to failure for a hard drive, or the forecasted number of calls to a call center on a given day or time. If you do not choose a numeric column, you might get an error.

    • If you do not specify which label column to use, Azure Machine Learning will try to infer which is the appropriate label column, by using the metadata of the dataset. If it picks the wrong column, use the column selector to correct it.


    If you have trouble using the Column Selector, see the article Select Columns in Dataset for tips. It describes some common scenarios and tips for using the WITH RULES and BY NAME options.

  4. Submit the pipeline. If you have a lot of data, it can take a while.


    If you have an ID column which is the ID of each row, or a text column, which contains too many unique values, Train Model may hit an error like "Number of unique values in column: "{column_name}" is greater than allowed.

    This is because the column hit the threshold of unique values, and may cause out of memory. You can use Edit Metadata to mark that column as Clear feature and it will not be used in training, or Extract N-Gram Features from Text module to preprocess text column. See Designer error code for more error details.

Model Interpretability

Model interpretability provides possibility to comprehend the ML model and to present the underlying basis for decision-making in a way that is understandable to humans.

Currently Train Model module supports using interpretability package to explain ML models. Following built-in algorithms are supported:

  • Linear Regression
  • Neural Network Regression
  • Boosted Decistion Tree Regression
  • Decision Forest Regression
  • Poisson Regression
  • Two-Class Logistic Regression
  • Two-Class Support Vector Machine
  • Two-Class Boosted Decistion Tree
  • Two-Class Decision Forest
  • Multi-class Decision Forest
  • Multi-class Logistic Regression
  • Multi-class Neural Network

To generate model explanations, you can select True in the drop-down list of Model Explanation in Train Model module. By default it is set to False in the Train Model module. Please note that generating explanation requires extra compute cost.

Screenshot showing model explanation checkbox

After the pipeline run completed, you can visit Explanations tab in the right pane of Train Model module, and explore the model performance, dataset and feature importance.

Screenshot showing model explanation charts

To learn more about using model explanations in Azure Machine Learning, refer to the how-to article about Interpret ML models.


After the model is trained:

  • To use the model in other pipelines, select the module and select the Register dataset icon under the Outputs tab in right panel. You can access saved models in the module palette under Datasets.

  • To use the model in predicting new values, connect it to the Score Model module, together with new input data.

Next steps

See the set of modules available to Azure Machine Learning.