lightgbm 0.1.5 → 0.1.6

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: dad09a50916e229119d99de66aa676868224af44d8958916284718edb7fb3a3f
-  data.tar.gz: 2f7d299a7037455a4ebd1ccaf41bc102ec66808b9e806a59194fd14699204364
+  metadata.gz: 7c609f529612c4aef7c34ecd59bb1b12df1d2b9bfb64ddae8e8a1eb4f5a8a2af
+  data.tar.gz: c0d1d48a5301dd66d5df8feee81e64140eb058dd1c38b1fccd8564b125d22aab
 SHA512:
-  metadata.gz: aa415946a82be07f1b9b8823b60fc75030f9e5547859078459583fa64d9212aad11db39d30e8e327b1fe23f48fc2ce9c6ddbd0445b52f73b44374a07c4724f5c
-  data.tar.gz: 99357888fde412672ecd804396cbd3a524b7dee6318d46d92a07064b858cd5f62f144f6b680a25ae0f1ae567df7d4670fcf38ba706fa31cbdfe58fd79eabc62d
+  metadata.gz: 5c2b002aa6e97a4d12de0076999c07be4077b6510f0b6a29cc6eb5d5f5ec23f2387ae52e82be0ab36ec2b00e27a93b337985ce1595de2845396ff9997c3172d5
+  data.tar.gz: ab3e47fb5c973576fc01632a0c1bf7479245f3037f61041c6c8596bd82ca2d37e1979474153e4a99302c5b276e98b7ed7772e0e00db21dcc90d4360bd775ed62

data/CHANGELOG.md

@@ -1,3 +1,8 @@
+## 0.1.6
+
+- Updated LightGBM to 2.3.0
+- Fixed error with JRuby
+
 ## 0.1.5
 
 - Packaged LightGBM with gem

data/README.md

@@ -14,7 +14,7 @@ Add this line to your application’s Gemfile:
 gem 'lightgbm'
 ```
 
-LightGBM is packaged with the gem, so no need to install it separately. On Mac, also run:
+On Mac, also install OpenMP:
 
 ```sh
 brew install libomp

data/lib/lightgbm/booster.rb

@@ -40,7 +40,7 @@ module LightGBM
       out_len = ::FFI::MemoryPointer.new(:int64)
       out_str = ::FFI::MemoryPointer.new(:string, buffer_len)
       check_result FFI.LGBM_BoosterDumpModel(handle_pointer, start_iteration, num_iteration, buffer_len, out_len, out_str)
-      actual_len = out_len.read_int64
+      actual_len = read_int64(out_len)
       if actual_len > buffer_len
         out_str = ::FFI::MemoryPointer.new(:string, actual_len)
         check_result FFI.LGBM_BoosterDumpModel(handle_pointer, start_iteration, num_iteration, actual_len, out_len, out_str)
@@ -87,7 +87,7 @@ module LightGBM
       out_len = ::FFI::MemoryPointer.new(:int64)
       out_str = ::FFI::MemoryPointer.new(:string, buffer_len)
       check_result FFI.LGBM_BoosterSaveModelToString(handle_pointer, start_iteration, num_iteration, buffer_len, out_len, out_str)
-      actual_len = out_len.read_int64
+      actual_len = read_int64(out_len)
       if actual_len > buffer_len
         out_str = ::FFI::MemoryPointer.new(:string, actual_len)
         check_result FFI.LGBM_BoosterSaveModelToString(handle_pointer, start_iteration, num_iteration, actual_len, out_len, out_str)
@@ -137,7 +137,7 @@ module LightGBM
       out_len = ::FFI::MemoryPointer.new(:int64)
       out_result = ::FFI::MemoryPointer.new(:double, num_class * input.count)
       check_result FFI.LGBM_BoosterPredictForMat(handle_pointer, data, 0, input.count, input.first.count, 1, 0, num_iteration, params_str(params), out_len, out_result)
-      out = out_result.read_array_of_double(out_len.read_int64)
+      out = out_result.read_array_of_double(read_int64(out_len))
       out = out.each_slice(num_class).to_a if num_class > 1
 
       singular ? out.first : out
@@ -202,6 +202,11 @@ module LightGBM
       out.read_int
     end
 
+    # read_int64 not available on JRuby
+    def read_int64(ptr)
+      ptr.read_array_of_int64(1).first
+    end
+
     include Utils
   end
 end

data/lib/lightgbm/dataset.rb

@@ -103,11 +103,6 @@ module LightGBM
       check_result FFI.LGBM_DatasetSaveBinary(handle_pointer, filename)
     end
 
-    # not released yet
-    # def dump_text(filename)
-    #   check_result FFI.LGBM_DatasetDumpText(handle_pointer, filename)
-    # end
-
     def subset(used_indices, params: nil)
       # categorical_feature passed via params
       params ||= self.params
@@ -129,6 +124,10 @@ module LightGBM
 
     private
 
+    def dump_text(filename)
+      check_result FFI.LGBM_DatasetDumpText(handle_pointer, filename)
+    end
+
     def field(field_name)
       num_data = self.num_data
       out_len = ::FFI::MemoryPointer.new(:int)

data/lib/lightgbm/ffi.rb

@@ -26,7 +26,7 @@ module LightGBM
     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_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
     attach_function :LGBM_DatasetGetNumData, %i[pointer pointer], :int

data/lib/lightgbm/version.rb

@@ -1,3 +1,3 @@
 module LightGBM
-  VERSION = "0.1.5"
+  VERSION = "0.1.6"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: lightgbm
 version: !ruby/object:Gem::Version
-  version: 0.1.5
+  version: 0.1.6
 platform: ruby
 authors:
 - Andrew Kane
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2019-09-03 00:00:00.000000000 Z
+date: 2019-09-30 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: ffi
@@ -80,20 +80,6 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '0'
-- !ruby/object:Gem::Dependency
-  name: numo-narray
-  requirement: !ruby/object:Gem::Requirement
-    requirements:
-    - - ">="
-      - !ruby/object:Gem::Version
-        version: '0'
-  type: :development
-  prerelease: false
-  version_requirements: !ruby/object:Gem::Requirement
-    requirements:
-    - - ">="
-      - !ruby/object:Gem::Version
-        version: '0'
 description: 
 email: andrew@chartkick.com
 executables: []