Сначала мы оставим 20 % данных для тестирования, а затем разделим оставшиеся 80 % на обучающие и проверочные:
from sklearn.model_selection import train_test_split

df_train_full, df_test = train_test_split(df, test_size=0.2, random_state=11)
df_train, df_val = train_test_split(df_train_full, test_size=0.25,