From 997a61557abb2e8205294b2ec2a892b17b323a27 Mon Sep 17 00:00:00 2001 From: Matthew Honnibal Date: Tue, 31 Oct 2017 19:30:52 +0100 Subject: [PATCH] Add vectors.n_keys property --- spacy/vectors.pyx | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/spacy/vectors.pyx b/spacy/vectors.pyx index cb9f0c0e6..bde793ec5 100644 --- a/spacy/vectors.pyx +++ b/spacy/vectors.pyx @@ -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.