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,25 @@
+import implicit
+
+from .mf_implicit_model import MFImplicitModel
+
+
+class BPR(MFImplicitModel):
+    """"""
+    def __init__(self,
+                 latent_dim,
+                 reg_term,
+                 learning_rate,
+                 epochs,
+                 **kwargs):
+
+        super(BPR, self).__init__(latent_dim=latent_dim,
+                                  reg_term=reg_term,
+                                  learning_rate=learning_rate,
+                                  epochs=epochs)
+
+        self.model = implicit.bpr.BayesianPersonalizedRanking(
+            factors=self.latent_dim,
+            learning_rate=self.learning_rate,
+            regularization=self.reg_term,
+            iterations=self.epochs
+        )