Normalization in TrieClassify.

antononcube · Apr 9, 2018 · f244122 · f244122
1 parent 25a168f
commit f244122
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/TriesWithFrequencies.m b/TriesWithFrequencies.m
@@ -591,7 +591,8 @@ Mathematica is (C) Copyright 1988-2016 Wolfram Research, Inc.
     Block[{res, dval = OptionValue[TrieClassify, "Default"]},
       res = TrieRetrieve[tr, record];
       If[Length[res] == 0, <|dval -> 0|>,
-        ReverseSort[Association[Rule @@@ TrieLeafProbabilities[res]]]
+        res = ReverseSort[Association[Rule @@@ TrieLeafProbabilities[res]]];
+        res / Total[res]
       ]
     ];