Author: gsingers
Date: Fri Sep 9 20:06:49 2011
New Revision: 1167345
URL: http://svn.apache.org/viewvc?rev=1167345&view=rev
Log:
MAHOUT-802: make item id look ups optional
Modified:
mahout/trunk/core/src/main/java/org/apache/mahout/cf/taste/hadoop/item/AggregateAndRecommendReducer.java
Modified:
mahout/trunk/core/src/main/java/org/apache/mahout/cf/taste/hadoop/item/AggregateAndRecommendReducer.java
URL:
http://svn.apache.org/viewvc/mahout/trunk/core/src/main/java/org/apache/mahout/cf/taste/hadoop/item/AggregateAndRecommendReducer.java?rev=1167345&r1=1167344&r2=1167345&view=diff
==============================================================================
---
mahout/trunk/core/src/main/java/org/apache/mahout/cf/taste/hadoop/item/AggregateAndRecommendReducer.java
(original)
+++
mahout/trunk/core/src/main/java/org/apache/mahout/cf/taste/hadoop/item/AggregateAndRecommendReducer.java
Fri Sep 9 20:06:49 2011
@@ -189,7 +189,12 @@ public final class AggregateAndRecommend
while (recommendationVectorIterator.hasNext()) {
Vector.Element element = recommendationVectorIterator.next();
int index = element.index();
- long itemID = indexItemIDMap.get(index);
+ long itemID;
+ if (indexItemIDMap != null && indexItemIDMap.isEmpty() == false) {
+ itemID = indexItemIDMap.get(index);
+ } else { //we don't have any mappings, so just use the original
+ itemID = index;
+ }
if (itemsToRecommendFor == null || itemsToRecommendFor.contains(itemID))
{
float value = (float) element.get();
if (!Float.isNaN(value)) {
