from sklearn import datasets
import pandas as pd
whole_data = datasets.load_iris()
whole_data
print('The full description of the dataset:\n',whole_data['DESCR'])
x_axis = whole_data.data[:,2] # Petal Length
y_axis = whole_data.data[:, 3] # Petal Width
# Plotting
import matplotlib.pyplot as plt
plt.scatter(x_axis, y_axis, c=whole_data.target)
plt.title("Violet: Setosa, Green: Versicolor, Yellow: Virginica")
plt.show()
I have chosen petal width and petal length for visualization as they show high class correlation ... why just me?