explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-04-22 10:02:01 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fix InMemoryCandidate attribute names.

Browse Source
This commit is contained in:
Raphael Mitsch 2023-03-15 10:51:34 +01:00
parent 80fb0666b9
commit 830939ee64
4 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
spacy/kb/candidate.pxd
View File

@ -8,8 +8,8 @@ cdef class Candidate:
cdef class InMemoryCandidate(Candidate):
cdef readonly hash_t _entity_hash
cdef readonly hash_t _alias_hash
cpdef vector[float] _entity_vector
cdef float _prior_prob
cdef readonly InMemoryLookupKB _kb
cdef float _entity_freq
cdef hash_t _mention

4
spacy/kb/candidate.pyx
View File

@ -64,7 +64,7 @@ cdef class InMemoryCandidate(Candidate):
self._entity_vector = entity_vector
self._prior_prob = prior_prob
self._kb = kb
self._alias = alias_hash
self._alias_hash = alias_hash
self._entity_freq = entity_freq
@property
@ -84,7 +84,7 @@ cdef class InMemoryCandidate(Candidate):
@property
def alias(self) -> str:
"""RETURNS (str): Alias."""
return self._kb.vocab.strings[self._alias]
return self._kb.vocab.strings[self._alias_hash]
@property
def entity_id_(self) -> str:

6
spacy/tests/pipeline/test_entity_linker.py
View File

@ -473,7 +473,7 @@ def test_candidate_generation(nlp):
# test the content of the candidates
assert adam_ent_cands[0].entity_id_ == "Q2"
assert adam_ent_cands[0].mention == "adam"
assert adam_ent_cands[0].alias == "adam"
assert_almost_equal(adam_ent_cands[0].entity_freq, 12)
assert_almost_equal(adam_ent_cands[0].prior_prob, 0.9)
@ -566,7 +566,7 @@ def test_vocab_serialization(nlp):
assert len(candidates) == 1
assert candidates[0].entity_id == q2_hash
assert candidates[0].entity_id_ == "Q2"
assert candidates[0].mention == "adam"
assert candidates[0].alias == "adam"
with make_tempdir() as d:
mykb.to_disk(d / "kb")
@ -577,7 +577,7 @@ def test_vocab_serialization(nlp):
assert len(candidates) == 1
assert candidates[0].entity_id == q2_hash
assert candidates[0].entity_id_ == "Q2"
assert candidates[0].mention == "adam"
assert candidates[0].alias == "adam"
assert kb_new_vocab.get_vector("Q2") == [2]
assert_almost_equal(kb_new_vocab.get_prior_prob("Q2", "douglas"), 0.4)

4
spacy/tests/serialize/test_serialize_kb.py
View File

@ -74,13 +74,13 @@ def _check_kb(kb):
assert candidates[0].entity_id_ == "Q007"
assert 6.999 < candidates[0].entity_freq < 7.01
assert candidates[0].entity_vector == [0, 0, 7]
assert candidates[0].mention == "double07"
assert candidates[0].alias == "double07"
assert 0.899 < candidates[0].prior_prob < 0.901
assert candidates[1].entity_id_ == "Q17"
assert 1.99 < candidates[1].entity_freq < 2.01
assert candidates[1].entity_vector == [7, 1, 0]
assert candidates[1].mention == "double07"
assert candidates[1].alias == "double07"
assert 0.099 < candidates[1].prior_prob < 0.101