Разделение на обучающую и контрольную (тестовую) выборки

from sklearn.model_selection import train_test_split

array = df.values

Выбор первых 4 столбцов

X = array[:,0:4]

Выбор пятого столбца

y = array[:,4]

Разделение X и y на обучающую и контрольную выборки

X_train, X_validation, Y_train, Y_validation = train_test_split(X, y, test_size=0.20, random_state=1)