Uso de sklearn.model_selection para dividir un modelo

train_subsets.py

from sklearn.model_selection import train_test_split

'''
Documentación https://scikit-learn.org/stable/modules/generated/sklearn.model_selection.train_test_split.html
divide un dataframe (df):
  * 80% train
  * 10% validate
  * 10% test
'''

train, test_and_validate = train_test_split(df, test_size=0.2, random_state=42, stratify=df['class'])

test, validate = train_test_split(test_and_validate, test_size=0.5, random_state=42, stratify=test_and_validate['class'])