svn commit: r930229 - /lucene/mahout/trunk/core/src/main/java/org/apache/mahout/cf/taste/hadoop/ToItemPrefsMapper.java

[sr...@apache.org]

Author: srowen
Date: Fri Apr  2 10:21:31 2010
New Revision: 930229

URL: http://svn.apache.org/viewvc?rev=930229&view=rev
Log:
MAHOUT-359, I think: allow data without prefs as input

Modified:
    lucene/mahout/trunk/core/src/main/java/org/apache/mahout/cf/taste/hadoop/ToItemPrefsMapper.java

Modified: lucene/mahout/trunk/core/src/main/java/org/apache/mahout/cf/taste/hadoop/ToItemPrefsMapper.java
URL: http://svn.apache.org/viewvc/lucene/mahout/trunk/core/src/main/java/org/apache/mahout/cf/taste/hadoop/ToItemPrefsMapper.java?rev=930229&r1=930228&r2=930229&view=diff
==============================================================================
--- lucene/mahout/trunk/core/src/main/java/org/apache/mahout/cf/taste/hadoop/ToItemPrefsMapper.java (original)
+++ lucene/mahout/trunk/core/src/main/java/org/apache/mahout/cf/taste/hadoop/ToItemPrefsMapper.java Fri Apr  2 10:21:31 2010
@@ -36,7 +36,8 @@ import org.apache.hadoop.mapred.Reporter
  * </p>
  * 
  * <p>
- * Each line is assumed to be of the form <code>userID,itemID,preference</code>.
+ * Each line is assumed to be of the form <code>userID,itemID,preference</code>, or
+ * <code>userID,itemID</code>.</p>
  * </p>
  * 
  * <h1>Output</h1>
@@ -59,7 +60,7 @@ public final class ToItemPrefsMapper ext
     String[] tokens = ToItemPrefsMapper.COMMA.split(value.toString());
     long userID = Long.parseLong(tokens[0]);
     long itemID = Long.parseLong(tokens[1]);
-    float prefValue = Float.parseFloat(tokens[2]);
+    float prefValue = tokens.length > 2 ? Float.parseFloat(tokens[2]) : 1.0f;
     output.collect(new LongWritable(userID), new ItemPrefWritable(itemID, prefValue));
   }