Update recom_ncf_base.py

PreferredAI · Oct 23, 2023 · 268d907 · 268d907
1 parent ebce517
commit 268d907
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/cornac/models/ncf/recom_ncf_base.py b/cornac/models/ncf/recom_ncf_base.py
@@ -178,8 +178,8 @@ def _fit_tf(self):
                         batch_users, batch_items, batch_ratings
                     ),
                 )
-                count += len(batch_ratings)
-                sum_loss += _loss * len(batch_ratings)
+                count += 1
+                sum_loss += _loss
                 if i % 10 == 0:
                     loop.set_postfix(loss=(sum_loss / count))