From 674de9cae3e18504b9801afadf47bc2f244e3d88 Mon Sep 17 00:00:00 2001 From: meirwahnon Date: Mon, 17 Jul 2017 16:01:49 +0300 Subject: [PATCH] change Probability order --- trees/id3.go | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/trees/id3.go b/trees/id3.go index 3730838..9043fec 100644 --- a/trees/id3.go +++ b/trees/id3.go @@ -291,7 +291,7 @@ func (o ClassesProba) Swap(i, j int) { o[i], o[j] = o[j], o[i] } func (o ClassesProba) Less(i, j int) bool { - return o[i].Probability < o[j].Probability + return o[i].Probability > o[j].Probability } // Predict class probabilities of the input samples what, returns a sorted array (by probability) of classes, and another array representing it's probabilities