Fix piskvorky#789 ignore DocvecsArray.doctag_syn0norm in save

accraze · Dec 22, 2016 · d58c09d · d58c09d
1 parent 1265809
commit d58c09d
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/gensim/models/doc2vec.py b/gensim/models/doc2vec.py
@@ -353,6 +353,11 @@ def __contains__(self, index):
         else:
             return index in self.doctags
 
+    def save(self, *args, **kwargs):
+        # don't bother storing the cached normalized vectors
+        kwargs['ignore'] = kwargs.get('ignore', ['syn0norm'])
+        super(DocvecsArray, self).save(*args, **kwargs)
+
     def borrow_from(self, other_docvecs):
         self.count = other_docvecs.count
         self.doctags = other_docvecs.doctags