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

Correction to randomisation and train-test split

Browse Source
This commit is contained in:
Richard Townsend 2014-05-18 11:23:32 +01:00
parent 26660e1470
commit 1b0e2dce7c
2 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
base/instances.go
View File

@ -175,6 +175,7 @@ func InstancesTrainTestSplit(src *Instances, prop float64) [2](*Instances) {
trainingRows := make([]int, 0) trainingRows := make([]int, 0)
testingRows := make([]int, 0) testingRows := make([]int, 0)
numAttrs := len(src.attributes) numAttrs := len(src.attributes)
src.Shuffle()
for i := 0; i < src.Rows; i++ { for i := 0; i < src.Rows; i++ {
trainOrTest := rand.Intn(101) trainOrTest := rand.Intn(101)
if trainOrTest > int(100*prop) { if trainOrTest > int(100*prop) {

2
ensemble/randomforest_test.go
View File

@ -13,7 +13,7 @@ func TestRandomForest1(testEnv *testing.T) {
if err != nil { if err != nil {
panic(err) panic(err)
} }
insts := base.InstancesTrainTestSplit(inst, 0.80) insts := base.InstancesTrainTestSplit(inst, 0.60)
filt := filters.NewChiMergeFilter(insts[0], 0.90) filt := filters.NewChiMergeFilter(insts[0], 0.90)
filt.AddAllNumericAttributes() filt.AddAllNumericAttributes()
filt.Build() filt.Build()