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)