Learn AI Today: 02 -- Introduction to Classification Problems using PyTorch

Let's get started by introducing the dataset. I will be using the very famous Iris flower dataset that contains 4 different measurements (sepal length, sepal width, petal length, petal width) of the following 3 species of flowers. The goal is to accurately identify the species using the 4 measurements for each flower. Note that nowadays it's relatively easy to use a model (Convolutional Neural Networks) that learns directly from the images but I will leave that topic for the next lesson. The Iris flower dataset can be easily downloaded from sklearn datasets as shown in the code below.