Spaces:

iSemantics
/

ner-demo-evaluate

Runtime error

elshehawy commited on Mar 4

Commit

c8acd87

•

1 Parent(s): df16c53

replace find_orgs function with find_orgs_in_data in evaluate_data.py

Files changed (1) hide show

evaluate_data.py CHANGED Viewed

@@ -115,7 +115,7 @@ def store_sample_data(data):
     for sent in data:
         labels = [ner_feature.feature.int2str(l) for l in sent['ner_tags']]
         # print(labels)
-        sent_orgs = find_orgs(sent['tokens'], labels)
         sent_text = tokenizer.convert_tokens_to_string(sent['tokens'])
         test_data.append({

     for sent in data:
         labels = [ner_feature.feature.int2str(l) for l in sent['ner_tags']]
         # print(labels)
+        sent_orgs = find_orgs_in_data(sent['tokens'], labels)
         sent_text = tokenizer.convert_tokens_to_string(sent['tokens'])
         test_data.append({