Add vectors.n_keys property

This commit is contained in:
Matthew Honnibal 2017-10-31 19:30:52 +01:00
parent 8075726838
commit 997a61557a
1 changed files with 6 additions and 1 deletions

View File

@ -77,6 +77,11 @@ cdef class Vectors:
"""Returns True if no keys are available for new keys."""
return len(self._unset) == 0
@property
def n_keys(self):
"""Returns True if no keys are available for new keys."""
return len(self.key2row)
def __reduce__(self):
keys_and_rows = self.key2row.items()
return (unpickle_vectors, (keys_and_rows, self.data))
@ -204,7 +209,7 @@ cdef class Vectors:
self._unset.remove(row)
return row
def most_similar(self, queries, *, return_scores=False, return_rows,
def most_similar(self, queries, *, return_scores=False, return_rows=False,
batch_size=1024):
'''For each of the given vectors, find the single entry most similar
to it, by cosine.