Training a Neural Network

Classification means placing an object into one of the possible categories, referred to as classes. For example, fruits are moving on a conveyor, and we want to sort them according to their type; apples must go in one basket, and oranges must go in another basket. Deciding if an image represents one type of fruit or the other is a classification task.

In this lesson, we collected images of 1000 objects from one of two categories: Rubik’s cubes and balls. A human annotated the dataset, i.e., they assigned the label 0 for a Rubik’s cube and 1 for a ball for each image. Let’s assume that we previously wrote a computer vision program to extract two features from each image.