Commit df8ac974 authored by vdberg's avatar vdberg
Browse files

crude ner added

parent 68719d65

srl.py

+3 −2
Original line number Diff line number Diff line
@@ -25,7 +25,7 @@ def crude_ner(sent): 
    global namlabs

    nampat = '\\b' + '\\b|\\b'.join([i for i in namlabs]) + '\\b'

    m = re.search(nampat, sent)

    return m

    return bool(m)





def ner(t):
@@ -45,6 +45,7 @@ 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")

vdberg_output.csv

0 → 100644
+1 −0
Original line number Diff line number Diff line 
,arg,verb,same