Commit f93e7417 authored by EstherMaria's avatar EstherMaria
Browse files
parents 860d2375 bba0d15b
Loading
Loading
Loading
Loading
+3 −1
Original line number Diff line number Diff line
@@ -46,9 +46,11 @@ parser.add_argument("-c", "--combine", action="store_true", default=False, help=
args = parser.parse_args()
df = load_output(dir='outputs', combine=args.combine)

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

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

print('Output size {}:'.format(df.shape))
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'])