This commit is contained in:
Jaakko Lukkari 2018-08-20 15:55:41 +03:00
Родитель 002eae5a0b
Коммит 2d11e50eee
1 изменённых файлов: 1 добавлений и 1 удалений

Просмотреть файл

@ -20,7 +20,7 @@ This library is meant for running predictions against a pre-trained XGBoost mode
// 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 column
// Make prediction for one row
res, _ := predictor.Predict(xgboost.FloatSliceVector([]float32{1, 2, 3}))
fmt.Printf("Results: %+v\n", res)
// output: Results: [1.08002]