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

Fix KeyError in Vectors.most_similar. Fixes #2648

Browse Source
This commit is contained in:
Matthew Honnibal 2018-12-10 16:19:18 +01:00
parent 90aec6d2f6
commit 449b889454
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
spacy/vectors.pyx
View File

@ -295,7 +295,8 @@ cdef class Vectors:
xp = get_array_module(self.data) xp = get_array_module(self.data)
row2key = {row: key for key, row in self.key2row.items()} row2key = {row: key for key, row in self.key2row.items()}
keys = xp.asarray([row2key[row] for row in best_rows], dtype='uint64') keys = xp.asarray(
[row2key[row] for row in best_rows if row in row2key], dtype='uint64')
return (keys, best_rows, scores) return (keys, best_rows, scores)
def from_glove(self, path): def from_glove(self, path):