explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2024-11-11 04:08:09 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fix load fail for missing vectors

Browse Source
This commit is contained in:
Matthew Honnibal 2017-08-19 22:07:00 +02:00
parent a3c51a0355
commit b8e1603cc4
1 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
spacy/vectors.pyx
View File

@ -32,8 +32,6 @@ cdef class Vectors:
self.data = data self.data = data
self.key2row = {} self.key2row = {}
self.keys = np.ndarray((self.data.shape[0],), dtype='uint64') self.keys = np.ndarray((self.data.shape[0],), dtype='uint64')
for string in strings:
self.add(string)
def __reduce__(self): def __reduce__(self):
return (Vectors, (self.strings, self.data)) return (Vectors, (self.strings, self.data))
@ -101,13 +99,15 @@ cdef class Vectors:
def from_disk(self, path, **exclude): def from_disk(self, path, **exclude):
def load_keys(path): def load_keys(path):
self.keys = numpy.load(path) if path.exists():
for i, key in enumerate(self.keys): self.keys = numpy.load(path)
self.keys[i] = key for i, key in enumerate(self.keys):
self.key2row[key] = i self.keys[i] = key
self.key2row[key] = i
def load_vectors(path): def load_vectors(path):
self.data = numpy.load(path) if path.exists():
self.data = numpy.load(path)
serializers = OrderedDict(( serializers = OrderedDict((
('keys', load_keys), ('keys', load_keys),