Roster Home Page Analysis
#Define Validation method #Train and validation set split from sklearn.model_selection import train_test_split X = df.drop('target_var', inplace=True, axis=1) y = df['column to predict'] X_train, X_val, y_train, y_val = train_test_split(X, y, test_size = 0.4, stratify = y.values, random_state = 101) #Cross validation from sklearn.model_selection import cross_val_score cross_val_score(model, X, y, cv=5) #StratifiedKFold from sklearn.model_selection import StratifiedKFold skf = StratifiedKFold(n_splits=5, random_state=101) for train_index, val_index in skf.split(X, y): X_train, X_val = X[train_index], X[val_index] y_train, y_val = y[train_index], y[val_index]