added patch from Low012 as submitted in

#256 (comment) which addresses a problem in the Bayesian Classifier source code as discussed in ptnplanet/Java-Naive-Bayes-Classifier#3
loklak · Mar 8, 2016 · 65ede16 · 65ede16
1 parent 1406fd5
commit 65ede16
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/org/loklak/objects/MessageEntry.java b/src/org/loklak/objects/MessageEntry.java
@@ -658,7 +658,8 @@ public Map<String, Object> toMap(final UserEntry user, final boolean calculatedD
                     assert c.getValue() != null;
                     if (c.getValue().getCategory() == Classifier.Category.NONE) continue; // we don't store non-existing classifications
                     m.put("classifier_" + c.getKey().name(), c.getValue().getCategory());
-                    m.put("classifier_" + c.getKey().name() + "_probability", c.getValue().getProbability());
+                    m.put("classifier_" + c.getKey().name() + "_probability",
+                        c.getValue().getProbability() == Float.POSITIVE_INFINITY ? Float.MAX_VALUE : c.getValue().getProbability());
                 }
             }