Commit ace41f1e authored by EstherMaria's avatar EstherMaria
Browse files
parents 7509a4e5 cedeefa8
Loading
Loading
Loading
Loading
+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'])