Commit 934cf368 authored by vdberg's avatar vdberg
Browse files

More prints

parent 019c78bd

srl.py

+3 −1
Original line number Diff line number Diff line
@@ -52,7 +52,7 @@ allen_ner = Predictor.from_path("https://s3-us-west-2.amazonaws.com/allennlp/mod 
verb_df = pd.DataFrame(columns=['arg', 'verb', 'same'])

for i,r in data.iterrows():

    names = ner(r.s)

    #print('Predicting NER for {}: {}'.format(i, r.s))

    print('Predicting NER for {}: {}'.format(i, r.s))



    roles = srl_pred.predict(sentence=r.s)['verbs']

    #print('roles', roles)
@@ -79,6 +79,8 @@ for i,r in data.iterrows(): 
        #print('verbs', verbs)

        verb_df = verb_df.append(verbs)

    print('Found {} verbs'.format(len(verbs)))

    print(verbs)

    print()



verb_df.arg = verb_df.arg.str[2:]

verb_df.to_csv('vdberg_output.csv')