In [8]:
# Spliting our data
X_train, X_test, y_train, y_test = split_data(X, y)