Commit cedeefa8 authored by vdberg's avatar vdberg
Browse files

set to all args

parent 6adef26d

process_srl.py

+3 −2
Original line number Diff line number Diff line
@@ -48,10 +48,11 @@ df = load_output(dir='outputs', combine=args.combine) 


print('Output size {}:'.format(df.shape))



df = df[df['arg'].str.startswith('B-ARG0')]

df = df[df['name'] != 'Ryan']



print('Arg-0 only {}:'.format(df.shape))

#df = df[df['arg'].str.startswith('B-ARG0')]

#print('Arg-0 only {}:'.format(df.shape))





sent = pd.read_csv('data/sent_df.csv', names=['sent_id','bias','doc_id','s'])

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