Here\'s the piece where I\'ve split the data :
X_train, X_test , y_train , y_test = train_test_split(data, labels ,test_size = 0.3, random_state = 1 ) prin