explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-08-03 11:50:19 +03:00
Code Issues Packages Projects Releases Wiki Activity

further rename

Browse Source
This commit is contained in:
svlandeg 2024-04-02 09:36:56 +02:00
parent 1a2c379e93
commit d7a250d994
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
spacy/pipeline/entity_linker.py
View File

@ -241,7 +241,7 @@ class EntityLinker(TrainablePipe):
if candidates_batch_size < 1: if candidates_batch_size < 1:
raise ValueError(Errors.E1044) raise ValueError(Errors.E1044)
def _score_augmented(examples: Iterable[Example], **kwargs): def _score_with_ents_set(examples: Iterable[Example], **kwargs):
# Because of how spaCy works, we can't just score immediately, because Language.evaluate # Because of how spaCy works, we can't just score immediately, because Language.evaluate
# calls pipe() on the predicted docs, which won't have entities if there is no NER in the pipeline. # calls pipe() on the predicted docs, which won't have entities if there is no NER in the pipeline.
if not self.use_gold_ents: if not self.use_gold_ents:
@ -255,7 +255,7 @@ class EntityLinker(TrainablePipe):
eg.predicted = doc eg.predicted = doc
return scorer(examples, **kwargs) return scorer(examples, **kwargs)
self.scorer = _score_augmented self.scorer = _score_with_ents_set
def _ensure_ents(self, examples: Iterable[Example]) -> Iterable[Example]: def _ensure_ents(self, examples: Iterable[Example]) -> Iterable[Example]:
"""If use_gold_ents is true, set the gold entities to (a copy of) eg.predicted.""" """If use_gold_ents is true, set the gold entities to (a copy of) eg.predicted."""
@ -401,9 +401,9 @@ class EntityLinker(TrainablePipe):
return losses return losses
def get_loss(self, examples: Iterable[Example], sentence_encodings: Floats2d): def get_loss(self, examples: Iterable[Example], sentence_encodings: Floats2d):
"""Here, we assume that get_loss is called with augmented examples if need be"""
validate_examples(examples, "EntityLinker.get_loss") validate_examples(examples, "EntityLinker.get_loss")
entity_encodings = [] entity_encodings = []
# We assume that get_loss is called with gold ents set in the examples if need be
eidx = 0 # indices in gold entities to keep eidx = 0 # indices in gold entities to keep
keep_ents = [] # indices in sentence_encodings to keep keep_ents = [] # indices in sentence_encodings to keep