Fixed issues with default vectorizer

x-tabdeveloping · Feb 1, 2024 · 652a933 · 652a933
1 parent 091533c
commit 652a933
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 6 deletions.
diff --git a/turftopic/models/decomp.py b/turftopic/models/decomp.py
@@ -46,7 +46,7 @@ def __init__(
         else:
             self.encoder_ = encoder
         if vectorizer is None:
-            self.vectorizer = default_vectorizer
+            self.vectorizer = default_vectorizer()
         else:
             self.vectorizer = vectorizer
         self.objective = objective

diff --git a/turftopic/models/gmm.py b/turftopic/models/gmm.py
@@ -47,20 +47,19 @@ def __init__(
             Encoder, str
         ] = "sentence-transformers/all-MiniLM-L6-v2",
         vectorizer: Optional[CountVectorizer] = None,
-        weight_prior: Literal[
-            "dirichlet", "dirichlet_process", None
-        ] = "dirichlet",
+        weight_prior: Literal["dirichlet", "dirichlet_process", None] = None,
         gamma: Optional[float] = None,
     ):
         self.n_components = n_components
         self.encoder = encoder
         self.weight_prior = weight_prior
+        self.gamma = gamma
         if isinstance(encoder, str):
             self.encoder_ = SentenceTransformer(encoder)
         else:
             self.encoder_ = encoder
         if vectorizer is None:
-            self.vectorizer = default_vectorizer
+            self.vectorizer = default_vectorizer()
         else:
             self.vectorizer = vectorizer
         if self.weight_prior is not None:

diff --git a/turftopic/models/keynmf.py b/turftopic/models/keynmf.py
@@ -82,7 +82,7 @@ def __init__(
         else:
             self.encoder_ = encoder
         if vectorizer is None:
-            self.vectorizer = default_vectorizer
+            self.vectorizer = default_vectorizer()
         else:
             self.vectorizer = vectorizer
         self.dict_vectorizer_ = DictVectorizer()