Изменения

X = ... <font color = green># загрузка X</font>

y = ... <font color = green># загрузка y</font>

<font color = green># разделение данных на train и test</font>

train_X, test_X, train_y, test_y = train_test_split(X, y)