Fix clear_vectors() method on Vocab

This commit is contained in:
Explosion Bot 2017-10-30 16:09:04 +01:00
parent 7b56b2f04b
commit aa64031751
1 changed files with 4 additions and 4 deletions

View File

@ -239,13 +239,13 @@ cdef class Vocab:
def vectors_length(self): def vectors_length(self):
return self.vectors.data.shape[1] return self.vectors.data.shape[1]
def clear_vectors(self, new_dim=None): def clear_vectors(self, width=None):
"""Drop the current vector table. Because all vectors must be the same """Drop the current vector table. Because all vectors must be the same
width, you have to call this to change the size of the vectors. width, you have to call this to change the size of the vectors.
""" """
if new_dim is None: if width is None:
new_dim = self.vectors.data.shape[1] width = self.vectors.data.shape[1]
self.vectors = Vectors(self.strings, width=new_dim) self.vectors = Vectors(self.strings, width=width)
def get_vector(self, orth): def get_vector(self, orth):
"""Retrieve a vector for a word in the vocabulary. Words can be looked """Retrieve a vector for a word in the vocabulary. Words can be looked