add_documents fix

ddangelov · Jun 23, 2021 · 5ce24bb · 5ce24bb
1 parent 8b55182
commit 5ce24bb
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/top2vec/Top2Vec.py b/top2vec/Top2Vec.py
@@ -1245,6 +1245,7 @@ def add_documents(self, documents, doc_ids=None, tokenizer=None, use_embedding_m
             self.model.docvecs.vectors_docs_norm = None
             self._set_document_vectors(np.vstack([self._get_document_vectors(norm=False), document_vectors]))
             self.model.docvecs.init_sims()
+            document_vectors = self._l2_normalize(document_vectors)
 
         else:
             if use_embedding_model_tokenizer: