Commit b57ddcec authored by EstherMaria's avatar EstherMaria
Browse files
parents 26dbf9aa c7d5dc1a
Loading
Loading
Loading
Loading
+2 −2
Original line number Diff line number Diff line
@@ -51,7 +51,7 @@ srl_pred = Predictor.from_path("https://s3-us-west-2.amazonaws.com/allennlp/mode
allen_ner = Predictor.from_path("https://s3-us-west-2.amazonaws.com/allennlp/models/ner-model-2018.12.18.tar.gz")

times = []
verb_df = pd.DataFrame(columns=['arg', 'verb', 'same'])
verb_df = pd.DataFrame(columns=['arg', 'verb', 'same', 'id'])
for i,r in data.iterrows():
    t0 = time.time()

@@ -81,7 +81,7 @@ for i,r in data.iterrows():
        #names['nambias'] = namlabs_df.loc[verbs.index].bias.values
        verbs['same'] = namlabs_df.loc[verbs.index].bias.values == r.bias
        #print('verbs', verbs)
        verbs['s_id'] = [i] * len(verbs)
        verbs['id'] = [i] * len(verbs)
        verb_df = verb_df.append(verbs)
        print('Found {} verbs'.format(len(verbs)))
        print(verbs)