You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@mahout.apache.org by sr...@apache.org on 2011/05/17 13:09:07 UTC

svn commit: r1104139 - /mahout/trunk/core/src/main/java/org/apache/mahout/cf/taste/impl/recommender/svd/AbstractFactorizer.java

Author: srowen
Date: Tue May 17 11:09:06 2011
New Revision: 1104139

URL: http://svn.apache.org/viewvc?rev=1104139&view=rev
Log:
MAHOUT-667 additional bug fix

Modified:
    mahout/trunk/core/src/main/java/org/apache/mahout/cf/taste/impl/recommender/svd/AbstractFactorizer.java

Modified: mahout/trunk/core/src/main/java/org/apache/mahout/cf/taste/impl/recommender/svd/AbstractFactorizer.java
URL: http://svn.apache.org/viewvc/mahout/trunk/core/src/main/java/org/apache/mahout/cf/taste/impl/recommender/svd/AbstractFactorizer.java?rev=1104139&r1=1104138&r2=1104139&view=diff
==============================================================================
--- mahout/trunk/core/src/main/java/org/apache/mahout/cf/taste/impl/recommender/svd/AbstractFactorizer.java (original)
+++ mahout/trunk/core/src/main/java/org/apache/mahout/cf/taste/impl/recommender/svd/AbstractFactorizer.java Tue May 17 11:09:06 2011
@@ -62,7 +62,8 @@ public abstract class AbstractFactorizer
   protected Integer userIndex(long userID) {
     Integer userIndex = userIDMapping.get(userID);
     if (userIndex == null) {
-      userIndex = userIDMapping.put(userID, userIDMapping.size());
+      userIndex = userIDMapping.size();
+      userIDMapping.put(userID, userIndex);
     }
     return userIndex;
   }
@@ -70,7 +71,8 @@ public abstract class AbstractFactorizer
   protected Integer itemIndex(long itemID) {
     Integer itemIndex = itemIDMapping.get(itemID);
     if (itemIndex == null) {
-      itemIndex = itemIDMapping.put(itemID, itemIDMapping.size());
+      itemIndex = itemIDMapping.size();
+      itemIDMapping.put(itemID, itemIndex);
     }
     return itemIndex;
   }