revert accidental changes to decomp models

x-tabdeveloping · Mar 20, 2024 · 5f8a999 · 5f8a999
1 parent 46944fa
commit 5f8a999
Showing 1 changed file with 0 additions and 8 deletions.
diff --git a/turftopic/models/decomp.py b/turftopic/models/decomp.py
@@ -47,7 +47,6 @@ def __init__(
         ] = "sentence-transformers/all-MiniLM-L6-v2",
         vectorizer: Optional[CountVectorizer] = None,
         objective: Literal["orthogonality", "independence"] = "independence",
-        subset: str = None
     ):
         self.n_components = n_components
         self.encoder = encoder
@@ -64,7 +63,6 @@ def __init__(
             self.decomposition = FastICA(n_components)
         else:
             self.decomposition = PCA(n_components)
-        self.subset = subset
 
     def fit_transform(
         self, raw_documents, y=None, embeddings: Optional[np.ndarray] = None
@@ -90,12 +88,6 @@ def fit_transform(
             console.log("Model fitting done.")
         return doc_topic
 
-    def abs_components(self, subset):
-        if subset == 'abs':
-            self.components_ = np.abs(self.components_)
-        if subset == 'neg':
-            self.components_ = -self.components_
-
     def transform(
         self, raw_documents, embeddings: Optional[np.ndarray] = None
     ) -> np.ndarray: