Commit d91f75b8 authored by vdberg's avatar vdberg
Browse files

fix

parent fc7f14e1

process_srl.py

+2 −0
Original line number Diff line number Diff line
@@ -75,6 +75,8 @@ output = output[output['name'] != 'Ryan'] 
#print('Arg-0 only {}:'.format(df.shape))



# combine output

output['sent_id'] = output['sent_id'].astype(int)

sent['sent_id'] = sent['sent_id'].astype(int)

df = pd.merge(output, sent, on=['sent_id'])

namlabs_df, namlabs = load_nameslab()

df = pd.merge(df, namlabs_df, on=['name'])