testing something

MaartenGr · Jul 6, 2024 · 4a9d7ea · 4a9d7ea
1 parent 69a20a9
commit 4a9d7ea
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/bertopic/vectorizers/_ctfidf.py b/bertopic/vectorizers/_ctfidf.py
@@ -85,6 +85,8 @@ def fit(self, X: sp.csr_matrix, multiplier: np.ndarray = None):
             if multiplier is not None:
                 idf = idf * multiplier
 
+            self.idf_ = idf
+
             self._idf_diag = sp.diags(
                 idf,
                 offsets=0,