Commit bba0d15b authored by vdberg's avatar vdberg
Browse files

debugging

parent 27b2e3bc
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'])