Main Content

Train Neural Network Classifiers Using Classification Learner App

This example shows how to create and compare neural network classifiers in the Classification Learner app, and export trained models to the workspace to make predictions for new data.

  1. In the MATLAB® Command Window, load the fisheriris data set, and create a table from the variables in the data set to use for classification.

    fishertable = readtable("fisheriris.csv");
    
  2. Click the Apps tab, and then click the Show more arrow on the right to open the apps gallery. In the Machine Learning and Deep Learning group, click Classification Learner.

  3. On the Learn tab, in the File section, click New Session and select From Workspace.

    Learn tab

  4. In the New Session from Workspace dialog box, select the table fishertable from the Data Set Variable list (if necessary). Observe that the app has selected response and predictor variables based on their data types. Petal and sepal length and width are predictors, and species is the response that you want to classify. For this example, do not change the selections.

  5. To accept the default validation scheme and continue, click Start Session. The default validation option is 5-fold cross-validation, to protect against overfitting.

    Classification Learner creates a scatter plot of the data.

  6. Use the scatter plot to investigate which variables are useful for predicting the response. Select different options in the X and Y lists under Predictors to visualize the distribution of species and measurements. Note which variables separate the species colors most clearly.

  7. Create a selection of neural network models. On the Learn tab, in the Models section, click the arrow to open the gallery. In the Neural Network Classifiers group, click All Neural Networks.

  8. In the Train section, click Train All and select Train All.

    Note

    • If you have Parallel Computing Toolbox™, then the Use Parallel button is selected by default. After you click Train All and select Train All or Train Selected, the app opens a parallel pool of workers. During this time, you cannot interact with the software. After the pool opens, you can continue to interact with the app while models train in parallel.

    • If you do not have Parallel Computing Toolbox, then the Use Background Training check box in the Train All menu is selected by default. After you select an option to train models, the app opens a background pool. After the pool opens, you can continue to interact with the app while models train in the background.

    Classification Learner trains one of each neural network classification option in the gallery, as well as the default fine tree model. In the Models pane, the app outlines the Accuracy (Validation) score of the best model. Classification Learner also displays a validation confusion matrix for the first neural network model (Narrow Neural Network).

  9. Select a model in the Models pane to view the results. For example, double-click the Narrow Neural Network model (model 2.1). Inspect the model Summary tab, which displays the Training Results and Additional Training Results metrics, calculated on the validation set.

  10. Examine the scatter plot for the trained model. On the Learn tab, in the Plots and Results section, click the arrow to open the gallery, and then click Scatter in the Validation Results group. Correctly classified points are marked with an O, and incorrectly classified points are marked with an X.

    Scatter plot of the Fisher iris data modeled by a neural network classifier

    Note

    Validation introduces some randomness into the results. Your model validation results can vary from the results shown in this example.

  11. Inspect the accuracy of the predictions in each class. On the Learn tab, in the Plots and Results section, click the arrow to open the gallery, and then click Confusion Matrix (Validation) in the Validation Results group. View the matrix of true class and predicted class results.

  12. For each remaining model, select the model in the Models pane, open the validation confusion matrix, and then compare the results across the models.

  13. Choose the best model in the Models pane (the best score is highlighted in the Accuracy (Validation) box). See if you can improve the model by removing features with low predictive power.

    First, duplicate the best model by right-clicking the model and selecting Duplicate.

  14. Investigate features to include or exclude using one of these methods.

    • Use the parallel coordinates plot. On the Learn tab, in the Plots and Results section, click the arrow to open the gallery, and then click Parallel Coordinates in the Validation Results group. Keep predictors that separate classes well.

      In the model Summary tab, you can specify the predictors to use during training. Click Feature Selection to expand the section, and specify predictors to remove from the model.

    • Use a feature ranking algorithm. On the Learn tab, in the Options section, click Feature Selection. In the Default Feature Selection tab, specify the feature ranking algorithm you want to use, and the number of features to keep among the highest ranked features. The bar graph can help you decide how many features to use.

      Click Save and Apply to save your changes. The new feature selection is applied to the existing draft model in the Models pane and will be applied to new draft models that you create using the gallery in the Models section of the Learn tab.

  15. Train the model. On the Learn tab, in the Train section, click Train All and select Train Selected to train the model using the new options. Compare results among the classifiers in the Models pane.

  16. Choose the best model in the Models pane. To try to improve the model further, change its hyperparameters. First, duplicate the best model by right-clicking the model and selecting Duplicate. Then, try changing hyperparameter settings, like the sizes of the fully connected layers or the regularization strength, in the model Summary tab. Train the new model by clicking Train All and selecting Train Selected in the Train section.

    To learn more about neural network model settings, see Neural Network Classifiers.

  17. You can export a full or compact version of the trained model to the workspace. On the Learn tab, in the Export section, click Export Model and select Export Model. To exclude the training data and export a compact model, clear the check box in the Export Classification Model dialog box. You can still use the compact model for making predictions on new data. In the dialog box, click OK to accept the default variable name.

  18. To examine the code for training this classifier, on the Learn tab, in the Export section, click Export Model and select Generate Function.

Tip

Use the same workflow to evaluate and compare the other classifier types you can train in Classification Learner.

To train all the nonoptimizable classifier model presets available for your data set:

  1. On the Learn tab, in the Models section, click the arrow to open the gallery of models.

  2. In the Get Started group, click All.

    Option selected for training all available classifier types

  3. In the Train section, click Train All and select Train All.

To learn about other classifier types, see Train Classification Models in Classification Learner App.

Related Topics