if the classifier has not been trained yet, there is no category the …

…classifier can use to classify
rkroll · Mar 11, 2012 · 6703a0c · 6703a0c
1 parent fd3577c
commit 6703a0c
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/BayesClassifier.java b/BayesClassifier.java
@@ -96,7 +96,9 @@ public int compare(Entry<K, Float> o1, Entry<K, Float> o2) {
  public K classify(Collection<T> features) {
  SortedSet<Entry<K, Float>> probabilites =
  this.categoryProbabilities(features);
- return probabilites.last().getKey();
+ if (probabilites.size() > 0)
+ return probabilites.last().getKey();
+ return null;
  }
 
 }