ups...

rkroll · Mar 11, 2012 · 654df58 · 654df58
1 parent 789f9c9
commit 654df58
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/Classifier.java b/Classifier.java
@@ -330,7 +330,9 @@ public float featureWeighedAverage(T feature, K category,
  ? this.featureProbability(feature, category)
  : calculator.featureProbability(feature, category);
 
- int totals = this.totalFeatureCount.get(feature);
+ Integer totals = this.totalFeatureCount.get(feature);
+ if (totals == null)
+ totals = 0;
  return (weight * assumedProbability + totals * basicProbability)
  / (weight + totals);
  }