diff --git a/src/cltl/nlp/spacy_nlp.py b/src/cltl/nlp/spacy_nlp.py index 7815a43..a4eec76 100644 --- a/src/cltl/nlp/spacy_nlp.py +++ b/src/cltl/nlp/spacy_nlp.py @@ -46,7 +46,7 @@ def _analyze_entities(self, doc): type = EntityType.OBJECT if type: - entities.append(Entity(token.text, type, (token.idx, token.idx + len(token.text)))) + entities.append(Entity(token.lemma_.lower(), type, (token.idx, token.idx + len(token.text)))) predicates[head_id][token.dep_] = token.lemma_