Skip to content

Class Mappings for Keras

This topic shows you how to set class mappings and their effects. The topic builds on Getting Started for Keras.

The following steps are covered:

  • Create a class mapping for MNIST.
  • Set the class mapping to MissingLink's callback.


Go through Getting Started for Keras.


Ensure that you can successfully run the training script that resulted from integration with the MissingLink SDK. In the steps that follow below, the script is further developed to include class mappings.

Write code

  1. Create a class mapping for MNIST.

    Right above the original setting of display name and description, add a mnist class mapping:

    mnist_class_mapping = {
        0: 'zero',
        1: 'one',
        2: 'two',
        3: 'three',
        4: 'four',
        5: 'five',
        6: 'six',
        7: 'seven',
        8: 'eight',
        9: 'nine',
  2. Set the class mapping to the callback.

    Modify the original set properties to the following:

        display_name='Keras convolutional neural network',
        description='Two dimensional convolutional neural network',

You should have added class mappings to your experiment successfully.

  • Inspect the resulting script here.

  • Run the new script and see how MissingLink's dashboard helps with monitoring the experiment's class mappings.