public code v1

2026-05-22 10:02:10 +02:00
commit 46a9ecf065
166 changed files with 6982454 additions and 0 deletions
@@ -0,0 +1,31 @@
+import implicit
+
+from .mf_implicit_model import MFImplicitModel
+
+
+class ALS(MFImplicitModel):
+    def __init__(
+        self,
+        latent_dim,
+        reg_term,
+        epochs,
+        random_state=42,
+        num_users=None,
+        num_items=None,
+        **kwargs,
+    ):
+        super(ALS, self).__init__(
+            latent_dim=latent_dim,
+            reg_term=reg_term,
+            epochs=epochs,
+            learning_rate=None,
+            num_users=num_users,
+            num_items=num_items,
+        )
+
+        self.model = implicit.als.AlternatingLeastSquares(
+            factors=self.latent_dim,
+            regularization=self.reg_term,
+            iterations=self.epochs,
+            random_state=random_state,
+        )