print("Full set:", iris_df.shape)

print("Train set:", X_train.shape, Y_train.shape)

print("Test set:", X_test.shape, Y_test.shape)