def merge_fea(tr_list, te_list): tr = loader.merge_fea(tr_list, primary_keys=ID_NAMES) te = loader.merge_fea(te_list, primary_keys=ID_NAMES) print(tr.head()) print(te.head()) loader.save_df(tr, tr_out_path) loader.save_df(te, te_out_path)
def merge_fea(tr_list, te_list): tr = loader.merge_fea(tr_list, primary_keys=ID_NAMES) te = loader.merge_fea(te_list, primary_keys=ID_NAMES) tr['impressions'] = tr['impressions'].astype('int') te['impressions'] = te['impressions'].astype('int') print(tr.head()) print(te.head()) print(tr[ID_NAMES].head()) loader.save_df(tr, tr_out_path) loader.save_df(te, te_out_path)