0 = Iris-setosa
1 = Iris-versicolor
2 = Iris-virginica