def load_test(): df_test = pd.read_csv(TEST, index_col='id') return extract_featurelist(df_test, features)
def load_train(features): df_train = pd.read_csv(TRAIN, index_col='id') return extract_featurelist(df_train, features)