explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-06-29 17:33:10 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fix candidate retrieval interface.

Browse Source
This commit is contained in:
Raphael Mitsch 2022-12-14 11:53:39 +01:00
parent 53a24abd8b
commit 51c485da09
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
spacy/kb/kb.pyx
View File

@ -41,16 +41,15 @@ cdef class KnowledgeBase:
RETURNS (Iterator[Iterable[Iterable[Candidate]]]): Identified candidates per document. RETURNS (Iterator[Iterable[Iterable[Candidate]]]): Identified candidates per document.
""" """
for doc in docs: for doc in docs:
yield [self.get_candidates(ent_span, doc) for ent_span in doc.ents] yield [self.get_candidates(ent_span) for ent_span in doc.ents]
def get_candidates(self, mention: Span, doc: Optional[Doc] = None) -> Iterable[Candidate]: def get_candidates(self, mention: Span) -> Iterable[Candidate]:
""" """
Return candidate entities for specified text. Each candidate defines the entity, the original alias, Return candidate entities for specified text. Each candidate defines the entity, the original alias,
and the prior probability of that alias resolving to that entity. and the prior probability of that alias resolving to that entity.
If the no candidate is found for a given text, an empty list is returned. If the no candidate is found for a given text, an empty list is returned.
Note that doc is not utilized for further context in this implementation. Note that doc is not utilized for further context in this implementation.
mention (Span): Mention for which to get candidates. mention (Span): Mention for which to get candidates.
doc (Optional[Doc]): Doc to use for context.
RETURNS (Iterable[Candidate]): Identified candidates. RETURNS (Iterable[Candidate]): Identified candidates.
""" """
raise NotImplementedError( raise NotImplementedError(