summaryrefslogtreecommitdiff
path: root/src/ner.py
diff options
context:
space:
mode:
Diffstat (limited to 'src/ner.py')
-rw-r--r--src/ner.py3
1 files changed, 3 insertions, 0 deletions
diff --git a/src/ner.py b/src/ner.py
index 2ca4afb..4ee788e 100644
--- a/src/ner.py
+++ b/src/ner.py
@@ -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