Refactor Word2VecSession fit method

owsky · Dec 18, 2023 · eabb273 · eabb273
1 parent 9bf3e70
commit eabb273
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/...nating_least_squares_9b566b411df48469660b513718ec895ccc1b678be56572b1ba08ea3ff6a201a6.job b/...nating_least_squares_9b566b411df48469660b513718ec895ccc1b678be56572b1ba08ea3ff6a201a6.job
diff --git a/src/models/hybrid/Word2VecSession.py b/src/models/hybrid/Word2VecSession.py
@@ -16,12 +16,12 @@ def __init__(self, data: Data):
     def fit(self, vector_size: int, window: int, biased: bool, silent=False):
         if not silent:
             print("Fitting Word2Vec model")
-        self.model = self._train_word2vec_model(vector_size, window, biased)
+        self.model = self._train_word2vec_model(vector_size, window)
         self.is_fit = True
         self.is_biased = biased
         return self
 
-    def _train_word2vec_model(self, vector_size: int, window: int, biased: bool):
+    def _train_word2vec_model(self, vector_size: int, window: int):
         n_users = self.data.interactions_train.shape[0]
         corpus = [self._get_str_liked(user_index) for user_index in range(n_users)]