diff options
| author | Shulhan <ms@kilabit.info> | 2019-03-02 04:59:38 +0700 |
|---|---|---|
| committer | Shulhan <ms@kilabit.info> | 2019-03-02 05:00:37 +0700 |
| commit | 8a5dfcccd447a9e2693bace81525d6264abccc63 (patch) | |
| tree | fa3d46ade2dd24cbd143343abfcb74ef83b0d3e0 /lib/mining/classifier/rf | |
| parent | 455e84bcebdf4731077b5949438b2f84fffa0e20 (diff) | |
| download | pakakeh.go-8a5dfcccd447a9e2693bace81525d6264abccc63.tar.xz | |
all: changes affected by refactoring on package "numbers"
Diffstat (limited to 'lib/mining/classifier/rf')
| -rw-r--r-- | lib/mining/classifier/rf/rf.go | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/lib/mining/classifier/rf/rf.go b/lib/mining/classifier/rf/rf.go index b6e08998..187612dd 100644 --- a/lib/mining/classifier/rf/rf.go +++ b/lib/mining/classifier/rf/rf.go @@ -18,9 +18,10 @@ import ( "math" "github.com/shuLhan/share/lib/debug" + "github.com/shuLhan/share/lib/floats64" + "github.com/shuLhan/share/lib/ints" "github.com/shuLhan/share/lib/mining/classifier" "github.com/shuLhan/share/lib/mining/classifier/cart" - "github.com/shuLhan/share/lib/numbers" libstrings "github.com/shuLhan/share/lib/strings" "github.com/shuLhan/share/lib/tabula" ) @@ -305,7 +306,7 @@ func (forest *Runtime) ClassifySet(samples tabula.ClasetInterface, // (1.2) classProbs := libstrings.FrequencyOfTokens(votes, vs, false) - _, idx, ok := numbers.Floats64FindMax(classProbs) + _, idx, ok := floats64.Max(classProbs) if ok { predicts = append(predicts, vs[idx]) @@ -346,7 +347,7 @@ func (forest *Runtime) Votes(sample *tabula.Row, sampleIdx int) ( for x, tree := range forest.trees { // (1) if sampleIdx >= 0 { - exist := numbers.IntsIsExist(forest.bagIndices[x], + exist := ints.IsExist(forest.bagIndices[x], sampleIdx) if exist { continue |