Commit f197dd1c authored by vdberg's avatar vdberg
Browse files

name recognition improved

parent 3fb31d4f

srl.py

+1 −1
Original line number Diff line number Diff line
@@ -71,7 +71,7 @@ for i,r in data.iterrows(): 
    #print('tagged', tagged)

    persons = tagged[tagged.n.str.endswith('PER')]

    #print('persons', persons)

    names = persons[persons.w.isin(namlabs)].copy()

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

    #print('names', names)



    verbs = names.set_index(['w']).iloc[:,1:].copy().stack().reset_index(level=1, name='arg').rename(columns={'level_1':'verb'})