bugfix

mokawi · Mar 17, 2021 · 05b12b8 · 05b12b8
1 parent 1c1d6df
commit 05b12b8
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/clctm.py b/clctm.py
@@ -144,7 +144,7 @@ def do_multiple(iids):
         self.unifs = set(self.input_ids)
 
         if vectorize:
-            self.vectorize(range(old_n_docs, self.n_docs))
+            self._vectorize_docs(range(old_n_docs, self.n_docs))
 
     def _get_indices(self, doc_idx):
         if isinstance(doc_idx, int):
@@ -187,6 +187,9 @@ def _vectorize_docs(self, start=0):
             self._get_single_doc(d) for d in tqdm.trange(start, self.n_docs, desc="Infering token vectors")
         ])
 
+    def vectorize_new(self):
+        self._vectorize_docs(len(self.token_vectors))
+
     def get_doc(self, doc_idx):
         """
         Returns the vector for document at position <doc_idx>. If `doc_idx` is a list, tuple or