lib/xgboost/ffi.rb in xgb-0.2.1 vs lib/xgboost/ffi.rb in xgb-0.3.0
- old
+ new
@@ -5,16 +5,17 @@
ffi_lib XGBoost.ffi_lib
# https://github.com/dmlc/xgboost/blob/master/include/xgboost/c_api.h
# keep same order
- # error
+ # general
+ attach_function :XGBoostVersion, %i[pointer pointer pointer], :void
attach_function :XGBGetLastError, %i[], :string
# dmatrix
attach_function :XGDMatrixCreateFromMat, %i[pointer uint64 uint64 float pointer], :int
- attach_function :XGDMatrixSetGroup, %i[pointer pointer uint64], :int
+ attach_function :XGDMatrixSetUIntInfo, %i[pointer string pointer uint64], :int
attach_function :XGDMatrixNumRow, %i[pointer pointer], :int
attach_function :XGDMatrixNumCol, %i[pointer pointer], :int
attach_function :XGDMatrixSliceDMatrix, %i[pointer pointer uint64 pointer], :int
attach_function :XGDMatrixFree, %i[pointer], :int
attach_function :XGDMatrixSaveBinary, %i[pointer string int], :int
@@ -25,10 +26,10 @@
attach_function :XGBoosterCreate, %i[pointer int pointer], :int
attach_function :XGBoosterUpdateOneIter, %i[pointer int pointer], :int
attach_function :XGBoosterEvalOneIter, %i[pointer int pointer pointer uint64 pointer], :int
attach_function :XGBoosterFree, %i[pointer], :int
attach_function :XGBoosterSetParam, %i[pointer string string], :int
- attach_function :XGBoosterPredict, %i[pointer pointer int int pointer pointer], :int
+ attach_function :XGBoosterPredict, %i[pointer pointer int int int pointer pointer], :int
attach_function :XGBoosterLoadModel, %i[pointer string], :int
attach_function :XGBoosterSaveModel, %i[pointer string], :int
attach_function :XGBoosterDumpModelEx, %i[pointer string int string pointer pointer], :int
attach_function :XGBoosterGetAttr, %i[pointer pointer pointer pointer], :int
attach_function :XGBoosterSetAttr, %i[pointer pointer pointer], :int