lib/lightgbm/ffi.rb in lightgbm-0.1.1 vs lib/lightgbm/ffi.rb in lightgbm-0.1.2
- old
+ new
@@ -10,10 +10,11 @@
attach_function :LGBM_GetLastError, %i[], :string
# dataset
attach_function :LGBM_DatasetCreateFromFile, %i[string string pointer pointer], :int
attach_function :LGBM_DatasetCreateFromMat, %i[pointer int int32 int32 int string pointer pointer], :int
+ attach_function :LGBM_DatasetGetSubset, %i[pointer pointer int32 string pointer], :int
attach_function :LGBM_DatasetFree, %i[pointer], :int
attach_function :LGBM_DatasetSaveBinary, %i[pointer string], :int
attach_function :LGBM_DatasetDumpText, %i[pointer string], :int
attach_function :LGBM_DatasetSetField, %i[pointer string pointer int int], :int
attach_function :LGBM_DatasetGetField, %i[pointer string pointer pointer pointer], :int
@@ -24,12 +25,18 @@
attach_function :LGBM_BoosterCreate, %i[pointer string pointer], :int
attach_function :LGBM_BoosterCreateFromModelfile, %i[string pointer pointer], :int
attach_function :LGBM_BoosterLoadModelFromString, %i[string pointer pointer], :int
attach_function :LGBM_BoosterFree, %i[pointer], :int
attach_function :LGBM_BoosterAddValidData, %i[pointer pointer], :int
+ attach_function :LGBM_BoosterGetNumClasses, %i[pointer pointer], :int
attach_function :LGBM_BoosterUpdateOneIter, %i[pointer pointer], :int
attach_function :LGBM_BoosterGetCurrentIteration, %i[pointer pointer], :int
+ attach_function :LGBM_BoosterNumModelPerIteration, %i[pointer pointer], :int
+ attach_function :LGBM_BoosterNumberOfTotalModel, %i[pointer pointer], :int
+ attach_function :LGBM_BoosterGetEvalCounts, %i[pointer pointer], :int
+ attach_function :LGBM_BoosterGetEvalNames, %i[pointer pointer pointer], :int
attach_function :LGBM_BoosterGetNumFeature, %i[pointer pointer], :int
+ attach_function :LGBM_BoosterGetEval, %i[pointer int pointer pointer], :int
attach_function :LGBM_BoosterPredictForMat, %i[pointer pointer int int32 int32 int int int string pointer pointer], :int
attach_function :LGBM_BoosterSaveModel, %i[pointer int int string], :int
attach_function :LGBM_BoosterSaveModelToString, %i[pointer int int int64 pointer pointer], :int
attach_function :LGBM_BoosterDumpModel, %i[pointer int int int64 pointer pointer], :int
attach_function :LGBM_BoosterFeatureImportance, %i[pointer int int pointer], :int