z4rd0s / go-xgboost

XGBoost bindings for golang

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

go-xgboost

Go bindings for XGBoost

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

Usage

This library is meant for running predictions against a pre-trained XGBoost model. Limited training related functionality is implemented 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