Fixed bug with embeddings

x-tabdeveloping · Aug 2, 2024 · 10d2711 · 10d2711
1 parent 7d98e12
commit 10d2711
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/turftopic/models/cluster.py b/turftopic/models/cluster.py
@@ -365,8 +365,8 @@ def fit_predict(
             if embeddings is None:
                 status.update("Encoding documents")
                 embeddings = self.encoder_.encode(raw_documents)
-                self.embeddings = embeddings
                 console.log("Encoding done.")
+            self.embeddings = embeddings
             status.update("Extracting terms")
             self.doc_term_matrix = self.vectorizer.fit_transform(raw_documents)
             console.log("Term extraction done.")