Commit 2a91448d authored by vdberg's avatar vdberg
Browse files

fix

parent bba0d15b

process_srl.py

+1 −0
Original line number Diff line number Diff line
@@ -49,6 +49,7 @@ 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))

srl.py

+1 −0
Original line number Diff line number Diff line
@@ -75,6 +75,7 @@ for i, r in data.iterrows(): 
    persons = tagged[tagged.n.str.endswith('PER')]

    names = persons[persons.w.apply(crude_ner) == True].copy()

    names.w = names.w.str.strip('—')

    names.w = names.w.str.strip(u'\ufeff')

    #print('names', names)



    verbs = names.set_index(['w']).iloc[:,1:]