DeepForge provides an example project for creating a classifier using the CIFAR10 dataset.
When first opening DeepForge in your browser (at http://localhost:8888 if following the instructions from the quick start), you will be prompted with a list of projects to open and provided the option to create a new project. For this example, let’s click “Create new…” and name our project “hello_cifar”.
Clicking “Create” will bring us to a prompt for the “seed” for our project. Select “cifar10” from the dropdown and click “Create”. This will now create our new project based on the cifar10 example provided with DeepForge.
In this example, we have three main pipelines:
download-normalize downloads and prepares our data. The
train pipeline trains a neural network model on the cifar10 dataset and the
test pipeline tests our trained model on our test set from the cifar10 dataset.
First, we will have to retrieve and prepare the data by running the
download-normalize pipeline. This can be done by opening the given pipeline then selecting the Execute Pipeline option from the action button in the lower right. As soon as that pipeline finishes, we can now use this data to train a neural network.
Next, we can open the
train pipeline. Before we execute the pipeline we have to set the input training data that we will be using. This is done by selecting the
Input operation then clicking the value for the
artifact field. This will provide all the possible options for the input data; for this example, we will want to select the “trainingdata” artifact. After setting the input, we can click on the
train operation to inspect the hyperparameters we are using and the architecture we are training. Selecting the
Output operation will allow you to change the name of the resulting artifact of this operation (in this case, a trained model). Finally, we can execute this pipeline like before to train the model.
As this operation trains, we can view the status by viewing the running execution. The easiest way to view the running execution is by clicking the given execution from the execution tray in the bottom left when viewing the originating pipeline.
Once the model has been trained, we can test the given model using the
test pipeline. In this pipeline, we have a few more inputs to set: “testing data”, “model to test” and the “human-readable class labels”. If you aren’t clear which operation provides which input, you can simply hover over it’s connected port on the
test operation. This will provide a tooltip with the full name of the input.
After setting the inputs for the
test pipeline (using the trained model and data from the first two pipelines), we can simply execute this pipeline to test our model. After executing the
test pipeline, we can view the execution and open the
test job to view the stdout for the given job. In the
test operation, this will allow us to view the printed accuracies of the model over each class.
And that’s it! We have just trained and tested our first neural network model using DeepForge. Although there are still a lot more advanced features that can be used, this should at least familiarize us with some of the core concepts in DeepForge.