mirror of
https://github.com/explosion/spaCy.git
synced 2025-08-02 11:20:19 +03:00
rename to _ensure_ents
This commit is contained in:
parent
ce7b51ef06
commit
1a2c379e93
|
@ -247,7 +247,7 @@ class EntityLinker(TrainablePipe):
|
|||
if not self.use_gold_ents:
|
||||
return scorer(examples, **kwargs)
|
||||
else:
|
||||
examples = self._augment_examples(examples)
|
||||
examples = self._ensure_ents(examples)
|
||||
docs = self.pipe(
|
||||
(eg.predicted for eg in examples),
|
||||
)
|
||||
|
@ -257,7 +257,7 @@ class EntityLinker(TrainablePipe):
|
|||
|
||||
self.scorer = _score_augmented
|
||||
|
||||
def _augment_examples(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 not self.use_gold_ents:
|
||||
return examples
|
||||
|
@ -311,7 +311,7 @@ class EntityLinker(TrainablePipe):
|
|||
nO = self.kb.entity_vector_length
|
||||
doc_sample = []
|
||||
vector_sample = []
|
||||
examples = self._augment_examples(islice(get_examples(), 10))
|
||||
examples = self._ensure_ents(islice(get_examples(), 10))
|
||||
for eg in examples:
|
||||
doc = eg.x
|
||||
doc_sample.append(doc)
|
||||
|
@ -379,7 +379,7 @@ class EntityLinker(TrainablePipe):
|
|||
losses.setdefault(self.name, 0.0)
|
||||
if not examples:
|
||||
return losses
|
||||
examples = self._augment_examples(examples)
|
||||
examples = self._ensure_ents(examples)
|
||||
validate_examples(examples, "EntityLinker.update")
|
||||
|
||||
# make sure we have something to learn from, if not, short-circuit
|
||||
|
|
Loading…
Reference in New Issue
Block a user