diff options
Diffstat (limited to 'src/ner.py')
| -rw-r--r-- | src/ner.py | 3 |
1 files changed, 3 insertions, 0 deletions
@@ -1,5 +1,7 @@ import spacy from collections import defaultdict +import pdb + class SpacyNER: def __init__(self,spacy_model): @@ -25,6 +27,7 @@ class SpacyNER: sentence_to_entities = defaultdict(list) unique_entities = set() passage_hash_id_to_entities = {} + pdb.set_trace() for ent in doc.ents: if ent.label_ == "ORDINAL" or ent.label_ == "CARDINAL": continue |
