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

printing less

parent f197dd1c

srl.py

+1 −3
Original line number Diff line number Diff line
@@ -45,7 +45,6 @@ sent = pd.read_csv('data/sent_df.csv') 
sent = sent[sent.s.apply(crude_ner) == True]

data = sent[:50]





srl_pred = Predictor.from_path("https://s3-us-west-2.amazonaws.com/allennlp/models/srl-model-2018.05.25.tar.gz")

allen_ner = Predictor.from_path("https://s3-us-west-2.amazonaws.com/allennlp/models/ner-model-2018.12.18.tar.gz")
@@ -54,7 +53,7 @@ logging.basicConfig(filename='vdberg_log.log',level=logging.DEBUG) 
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))

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



    roles = srl_pred.predict(sentence=r.s)['verbs']
@@ -83,7 +82,6 @@ for i,r in data.iterrows(): 
    verbs['same'] = namlabs_df.loc[verbs.index].bias.values == r.bias

    #print('verbs', verbs)

    verb_df = verb_df.append(verbs)

    print(verb_df)



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

verb_df.to_csv('vdberg_output.csv')