OrgLion-ML/golearn
1
0
Fork 0
mirror of https://github.com/sjwhitworth/golearn.git synced 2025-04-26 13:49:14 +08:00
Code Issues Releases Wiki Activity

change Probability order

Browse Source
This commit is contained in:
meirwahnon 2017-07-17 16:01:49 +03:00
parent 518c0d84c4
commit 674de9cae3
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
trees/id3.go
View File

@ -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