text_plain_output_1.png
from sklearn.model_selection import train_test_split

import matplotlib.pyplot as plt

import seaborn as sns

tmp = data.drop('Id', axis=1)
g = sns.pairplot(tmp, hue='Species', markers='+')
plt.show()

X = data.drop(['Id', 'Species'], axis=1)
y = data['Species']

X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.4, random_state=5)
print(X_train.shape)
print(y_train.shape)
print(X_test.shape)
print(y_test.shape)