mirror of
https://github.com/sjwhitworth/golearn.git
synced 2025-04-26 13:49:14 +08:00
21 lines
491 B
Go
21 lines
491 B
Go
|
package trees_test
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/sjwhitworth/golearn/base"
|
||
|
|
"github.com/sjwhitworth/golearn/ensemble"
|
||
|
|
"testing"
|
||
|
|
)
|
||
|
|
|
||
|
|
func BenchmarkRandomForestFit(b *testing.B) {
|
||
|
|
// benchdata.csv contains ../examples/datasets/iris.csv repeated 100 times.
|
||
|
|
data, err := base.ParseCSVToInstances("benchdata.csv", true)
|
||
|
|
if err != nil {
|
||
|
|
b.Fatalf("Cannot load benchdata.csv err:\n%v", err)
|
||
|
|
}
|
||
|
|
b.ResetTimer()
|
||
|
|
tree := ensemble.NewRandomForest(20, 4)
|
||
|
|
for i := 0; i < b.N; i++ {
|
||
|
|
tree.Fit(data)
|
||
|
|
}
|
||
|
|
}
|