Bug fix.

antononcube · Aug 19, 2019 · 1b746e3 · 1b746e3
1 parent 5b77318
commit 1b746e3
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/JavaTriesWithFrequencies.m b/JavaTriesWithFrequencies.m
@@ -522,7 +522,7 @@ Mathematica is (C) Copyright 1988-2017 Wolfram Research, Inc.
     Block[{res, dval = OptionValue[JavaTrieClassify, "Default"]},
       res = JavaTrieLeafProbabilities[JavaTrieRetrieve[tr, record], "Normalized"->True];
       If[Length[res] == 0, <|dval -> 0|>,
-        res = AssociationThread[res[[All, 2, 2]] -> res[[All, 1, 2]]];
+        res = AssociationThread[res[[All, 1, 2]] -> res[[All, 2, 2]]];
         res = ReverseSort[Association[Rule @@@ res]]
       ]
     ];
@@ -531,13 +531,13 @@ Mathematica is (C) Copyright 1988-2017 Wolfram Research, Inc.
     First @* Keys @* TakeLargest[1] /@ JavaTrieClassify[tr, records, "Probabilities", opts];
 
 JavaTrieClassify[tr_, records:(_Dataset|{_List..}), "Probability" -> class_, opts : OptionsPattern[]] :=
-    Map[Lookup[#, class, 0]&, JavaTrieClassify[tr, records, "Probabilities", opts] ];
+    Map[ Lookup[#, class, 0]&, JavaTrieClassify[tr, records, "Probabilities", opts] ];
 
 JavaTrieClassify[tr_, records:(_Dataset|{_List..}), "TopProbabilities", opts : OptionsPattern[]] :=
     Map[ Select[#, # > 0 &]&, JavaTrieClassify[tr, records, "Probabilities", opts] ];
 
 JavaTrieClassify[tr_, records:(_Dataset|{_List..}), "TopProbabilities" -> n_Integer, opts : OptionsPattern[]] :=
-    Map[TakeLargest[#, UpTo[n]]&, JavaTrieClassify[tr, records, "Probabilities", opts] ];
+    Map[ TakeLargest[#, UpTo[n]]&, JavaTrieClassify[tr, records, "Probabilities", opts] ];
 
 JavaTrieClassify[tr_, records:(_Dataset|{_List..}), "Probabilities", opts:OptionsPattern[] ] :=
     Block[{clRes, classLabels, stencil},