Bugfix for SVD stochastic gradient descent and RMSE calculation

chujiaqiang · Jan 16, 2012 · 040eedf · 040eedf
1 parent 7032dd5
commit 040eedf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scikits/crab/recommenders/svd/classes.py b/scikits/crab/recommenders/svd/classes.py
@@ -237,7 +237,7 @@ def _train(self, rating_indices, update_user, update_item):
             i_f = self.item_factors[item_idx]
 
             #Compute factor updates
-            delta_u = err * u_i - self.regularization * u_f
+            delta_u = err * i_f - self.regularization * u_f
             delta_i = err * u_f - self.regularization * i_f
             #if necessary apply updates
             if update_user: