Guide for Audio Classification Model with Marty the Robot
This guide explains how to create and use an audio classification model, and have Marty the Robot respond to predictions in MartyBlocks.
Navigate to the Machine Learning tab, press the “Add” button at the bottom left corner, and select “New Audio Model”.
If a new dialog box comes up asking you if you are sure, press continue.
You can start by giving your model a new name!
Type the new model name in the “Model” input at the top.
You can now start collecting data for your classes! As a general rule of thumb, the more data you give to the model the better will perform, although the quality of the data is equally important!
Just type the name of the class you want to collect data for in the “Collect Data for” input and press the “Record” button.
Repeat this process for all the classes you want to collect data for.
You can collect more samples for each class by pressing the “Record” button again.
Note that the audio model requires at least 10 samples for the Background Noise class.
Once you have collected enough data for your classes, you can train your model by pressing the “Train” button.
Once the training is complete, you will see the Loss graph which plots the training loss over training iterations (epochs).
Once you have trained your model, you can run it by pressing the “Run” button.
You will see the predictions for each class in the “Predictions” table.
Once you are happy with your model, you can save it by pressing the “Save” button.
Note that once your model is saved you will not be able to collect more data for it.
You can now load your model in MartyBlocks and use it to make predictions!
First, navigate to the Code tab and scroll down to the Machine Learning blocks section.
Use the “Load Marty Machine model” block and select the model you want to load.
Sound classifications should start automatically after importing the model. You may toggle classifications on and off using the “turn classification” block.
Once you have executed the command to import your classification model, the class labels will be available for selection in some blocks. There are a few ways to make predictions:
To see the value of the “sound label” variable, check the box beside the variable in the Teachable Machine menu.
To make changes to the rate of classifications, use the “Label once every seconds” block
To make changes to the confidence threshold, use the “set confidence threshold” block. The default confidence threshold is 0.5 (i.e. 50%).