pure Go implementation of prediction part for GBRT (Gradient Boosting Regression Trees) models from popular frameworks
place files from testdata_v3.zip
to testdata
folder to activate additional large scale test cases and benchmarks
to run tests:
cd $GOPATH/src/github.com/dmitryikh/leaves
go test -v
to run benchmarks:
cd $GOPATH/src/github.com/dmitryikh/leaves
go test -bench .
place files from testdata_v2.zip
to testdata
folder to activate additional large scale test cases and benchmarks
to run tests:
cd $GOPATH/src/github.com/dmitryikh/leaves
go test -v
to run benchmarks:
cd $GOPATH/src/github.com/dmitryikh/leaves
go test -bench .
place files from testdata.zip
to testdata
folder to activate additional large scale test cases and benchmarks
to run tests:
cd $GOPATH/src/github.com/dmitryikh/leaves
go test -v
to run benchmarks:
cd $GOPATH/src/github.com/dmitryikh/leaves
go test -bench .