ffi.rb in lightgbm-0.1.5

- old
+ new

@@ -4,10 +4,13 @@
 
     begin
       ffi_lib LightGBM.ffi_lib
     rescue LoadError => e
       raise e if ENV["LIGHTGBM_DEBUG"]
+      if e.message.include?("libomp")
+        raise LoadError, "Could not find OpenMP"
+      end
       raise LoadError, "Could not find LightGBM"
     end
 
     # https://github.com/microsoft/LightGBM/blob/master/include/LightGBM/c_api.h
     # keep same order
@@ -17,9 +20,11 @@
 
     # 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_DatasetSetFeatureNames, %i[pointer pointer int], :int
+    attach_function :LGBM_DatasetGetFeatureNames, %i[pointer pointer 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