This commit is contained in:
Jaakko Lukkari 2018-08-20 15:17:51 +03:00
Родитель 2de9a3d6bf
Коммит 032389edd3
1 изменённых файлов: 2 добавлений и 0 удалений

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

@ -73,6 +73,8 @@ func (booster *XGBooster) Predict(mat *XGDMatrix, optionMask int, ntreeLimit uin
sliceHeader.Len = int(outLen)
sliceHeader.Data = uintptr(unsafe.Pointer(outResult))
runtime.KeepAlive(mat)
return copyFloat32Slice(list), nil
}