8000 GitHub - Unity-Technologies/go-xgboost: XGBoost bindings for golang
[go: up one dir, main page]
More Web Proxy on the site http://driver.im/
Skip to content

Unity-Technologies/go-xgboost

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

33 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

WORK IN PROGRESS... USE AT OWN RISK :-)

Build Status GoDoc

go-xgboost

Go bindings for XGBoost

import "github.com/Applifier/go-xgboost"

Usage

This library is meant for running predictions against a pre-trained XGBoost model. Limited training related functionality is i 66FD mplemented under core but training the model in python or using the xgboost cli is encouraged.

// Create predictor for a model and define the number of workers (and other settings)
predictor, _ := xgboost.NewPredictor(modelPath, runtime.NumCPU(), 0, 0, -1)

// Make prediction for one row
res, _ := predictor.Predict(xgboost.FloatSliceVector([]float32{1, 2, 3}))
fmt.Printf("Results: %+v\n", res)
// output: Results: [1.08002]

License

MIT

0