lightgbm 0.1.6 → 0.1.7

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 7c609f529612c4aef7c34ecd59bb1b12df1d2b9bfb64ddae8e8a1eb4f5a8a2af
-  data.tar.gz: c0d1d48a5301dd66d5df8feee81e64140eb058dd1c38b1fccd8564b125d22aab
+  metadata.gz: cf13db584839dc0cb0a9b6d2480a6846704aee74f6a984300db06901cfbb1e66
+  data.tar.gz: ec716f3a2354c28e916b4ae89bd567e222f8bb5e5415e9579fb3c67def9c8edc
 SHA512:
-  metadata.gz: 5c2b002aa6e97a4d12de0076999c07be4077b6510f0b6a29cc6eb5d5f5ec23f2387ae52e82be0ab36ec2b00e27a93b337985ce1595de2845396ff9997c3172d5
-  data.tar.gz: ab3e47fb5c973576fc01632a0c1bf7479245f3037f61041c6c8596bd82ca2d37e1979474153e4a99302c5b276e98b7ed7772e0e00db21dcc90d4360bd775ed62
+  metadata.gz: db7d6da179096a0648b2257c575383e9a59caa6bae04f39275aafe9126e69b51c6048c1452d2fb8dffa538362614efe83f7c05a8df4c749808040835a977623c
+  data.tar.gz: 5ebecf54455d3aef8ff3235e196bc064353bb7379a8b6fa61671feb223813ac6d028d7b86a0c9ada0568b06d048c6fcfdea48f082d208edaecbfdb54ead75d33

data/CHANGELOG.md

@@ -1,16 +1,21 @@
-## 0.1.6
+## 0.1.7 (2019-12-05)
+
+- Updated LightGBM to 2.3.1
+- Switched to doubles for datasets and predictions
+
+## 0.1.6 (2019-09-29)
 
 - Updated LightGBM to 2.3.0
 - Fixed error with JRuby
 
-## 0.1.5
+## 0.1.5 (2019-09-03)
 
 - Packaged LightGBM with gem
 - Added support for missing values
 - Added `feature_names` to datasets
 - Fixed Daru training and prediction
 
-## 0.1.4
+## 0.1.4 (2019-08-19)
 
 - Friendlier message when LightGBM not found
 - Added `Ranker`
@@ -18,22 +23,22 @@
 - Free memory when objects are destroyed
 - Removed unreleased `dump_text` method
 
-## 0.1.3
+## 0.1.3 (2019-08-16)
 
 - Added Scikit-Learn API
 - Added support for Daru and Numo::NArray
 
-## 0.1.2
+## 0.1.2 (2019-08-15)
 
 - Added `cv` method
 - Added early stopping
 - Fixed multiclass classification
 
-## 0.1.1
+## 0.1.1 (2019-08-14)
 
 - Added training API
 - Added many methods
 
-## 0.1.0
+## 0.1.0 (2019-08-13)
 
 - First release

data/README.md

@@ -165,7 +165,7 @@ Numo::DFloat.new(3, 2).seq
 
 ## Credits
 
-Thanks to the [xgboost](https://github.com/PairOnAir/xgboost-ruby) gem for serving as an initial reference, and Selva Prabhakaran for the [test datasets](https://github.com/selva86/datasets).
+Thanks to the [xgboost](https://github.com/PairOnAir/xgboost-ruby) gem for serving as an initial reference.
 
 ## History
 
@@ -179,3 +179,13 @@ Everyone is encouraged to help improve this project. Here are a few ways you can
 - Fix bugs and [submit pull requests](https://github.com/ankane/lightgbm/pulls)
 - Write, clarify, or fix documentation
 - Suggest or add new features
+
+To get started with development and testing:
+
+```sh
+git clone https://github.com/ankane/lightgbm.git
+cd lightgbm
+bundle install
+bundle exec rake vendor:all
+bundle exec rake test
+```

data/lib/lightgbm/booster.rb

@@ -30,7 +30,7 @@ module LightGBM
 
     def current_iteration
       out = ::FFI::MemoryPointer.new(:int)
-      check_result FFI::LGBM_BoosterGetCurrentIteration(handle_pointer, out)
+      check_result FFI.LGBM_BoosterGetCurrentIteration(handle_pointer, out)
       out.read_int
     end
 
@@ -38,11 +38,11 @@ module LightGBM
       num_iteration ||= best_iteration
       buffer_len = 1 << 20
       out_len = ::FFI::MemoryPointer.new(:int64)
-      out_str = ::FFI::MemoryPointer.new(:string, buffer_len)
+      out_str = ::FFI::MemoryPointer.new(:char, buffer_len)
       check_result FFI.LGBM_BoosterDumpModel(handle_pointer, start_iteration, num_iteration, buffer_len, out_len, out_str)
       actual_len = read_int64(out_len)
       if actual_len > buffer_len
-        out_str = ::FFI::MemoryPointer.new(:string, actual_len)
+        out_str = ::FFI::MemoryPointer.new(:char, actual_len)
         check_result FFI.LGBM_BoosterDumpModel(handle_pointer, start_iteration, num_iteration, actual_len, out_len, out_str)
       end
       out_str.read_string
@@ -85,11 +85,11 @@ module LightGBM
       num_iteration ||= best_iteration
       buffer_len = 1 << 20
       out_len = ::FFI::MemoryPointer.new(:int64)
-      out_str = ::FFI::MemoryPointer.new(:string, buffer_len)
+      out_str = ::FFI::MemoryPointer.new(:char, buffer_len)
       check_result FFI.LGBM_BoosterSaveModelToString(handle_pointer, start_iteration, num_iteration, buffer_len, out_len, out_str)
       actual_len = read_int64(out_len)
       if actual_len > buffer_len
-        out_str = ::FFI::MemoryPointer.new(:string, actual_len)
+        out_str = ::FFI::MemoryPointer.new(:char, actual_len)
         check_result FFI.LGBM_BoosterSaveModelToString(handle_pointer, start_iteration, num_iteration, actual_len, out_len, out_str)
       end
       out_str.read_string
@@ -104,13 +104,13 @@ module LightGBM
 
     def num_model_per_iteration
       out = ::FFI::MemoryPointer.new(:int)
-      check_result FFI::LGBM_BoosterNumModelPerIteration(handle_pointer, out)
+      check_result FFI.LGBM_BoosterNumModelPerIteration(handle_pointer, out)
       out.read_int
     end
 
     def num_trees
       out = ::FFI::MemoryPointer.new(:int)
-      check_result FFI::LGBM_BoosterNumberOfTotalModel(handle_pointer, out)
+      check_result FFI.LGBM_BoosterNumberOfTotalModel(handle_pointer, out)
       out.read_int
     end
 
@@ -131,12 +131,12 @@ module LightGBM
 
       flat_input = input.flatten
       handle_missing(flat_input)
-      data = ::FFI::MemoryPointer.new(:float, input.count * input.first.count)
-      data.put_array_of_float(0, flat_input)
+      data = ::FFI::MemoryPointer.new(:double, input.count * input.first.count)
+      data.write_array_of_double(flat_input)
 
       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)
+      check_result FFI.LGBM_BoosterPredictForMat(handle_pointer, data, 1, input.count, input.first.count, 1, 0, num_iteration, params_str(params), out_len, out_result)
       out = out_result.read_array_of_double(read_int64(out_len))
       out = out.each_slice(num_class).to_a if num_class > 1
 
@@ -168,7 +168,7 @@ module LightGBM
 
     def eval_counts
       out = ::FFI::MemoryPointer.new(:int)
-      check_result FFI::LGBM_BoosterGetEvalCounts(handle_pointer, out)
+      check_result FFI.LGBM_BoosterGetEvalCounts(handle_pointer, out)
       out.read_int
     end
 
@@ -176,8 +176,8 @@ module LightGBM
       eval_counts ||= eval_counts()
       out_len = ::FFI::MemoryPointer.new(:int)
       out_strs = ::FFI::MemoryPointer.new(:pointer, eval_counts)
-      str_ptrs = eval_counts.times.map { ::FFI::MemoryPointer.new(:string, 255) }
-      out_strs.put_array_of_pointer(0, str_ptrs)
+      str_ptrs = eval_counts.times.map { ::FFI::MemoryPointer.new(:char, 255) }
+      out_strs.write_array_of_pointer(str_ptrs)
       check_result FFI.LGBM_BoosterGetEvalNames(handle_pointer, out_len, out_strs)
       str_ptrs.map(&:read_string)
     end
@@ -198,7 +198,7 @@ module LightGBM
 
     def num_class
       out = ::FFI::MemoryPointer.new(:int)
-      check_result FFI::LGBM_BoosterGetNumClasses(handle_pointer, out)
+      check_result FFI.LGBM_BoosterGetNumClasses(handle_pointer, out)
       out.read_int
     end
 

data/lib/lightgbm/dataset.rb

@@ -17,7 +17,7 @@ module LightGBM
       reference = reference.handle_pointer if reference
       if used_indices
         used_row_indices = ::FFI::MemoryPointer.new(:int32, used_indices.count)
-        used_row_indices.put_array_of_int32(0, used_indices)
+        used_row_indices.write_array_of_int32(used_indices)
         check_result FFI.LGBM_DatasetGetSubset(reference, used_row_indices, used_indices.count, parameters, @handle)
       elsif data.is_a?(String)
         check_result FFI.LGBM_DatasetCreateFromFile(data, parameters, reference, @handle)
@@ -39,9 +39,9 @@ module LightGBM
         end
 
         handle_missing(flat_data)
-        c_data = ::FFI::MemoryPointer.new(:float, nrow * ncol)
-        c_data.put_array_of_float(0, flat_data)
-        check_result FFI.LGBM_DatasetCreateFromMat(c_data, 0, nrow, ncol, 1, parameters, reference, @handle)
+        c_data = ::FFI::MemoryPointer.new(:double, nrow * ncol)
+        c_data.write_array_of_double(flat_data)
+        check_result FFI.LGBM_DatasetCreateFromMat(c_data, 1, nrow, ncol, 1, parameters, reference, @handle)
       end
       ObjectSpace.define_finalizer(self, self.class.finalize(handle_pointer)) unless used_indices
 
@@ -67,8 +67,8 @@ module LightGBM
       # must preallocate space
       num_feature_names = ::FFI::MemoryPointer.new(:int)
       out_strs = ::FFI::MemoryPointer.new(:pointer, 1000)
-      str_ptrs = 1000.times.map { ::FFI::MemoryPointer.new(:string, 255) }
-      out_strs.put_array_of_pointer(0, str_ptrs)
+      str_ptrs = 1000.times.map { ::FFI::MemoryPointer.new(:char, 255) }
+      out_strs.write_array_of_pointer(str_ptrs)
       check_result FFI.LGBM_DatasetGetFeatureNames(handle_pointer, out_strs, num_feature_names)
       str_ptrs[0, num_feature_names.read_int].map(&:read_string)
     end
@@ -141,11 +141,11 @@ module LightGBM
       data = data.to_a unless data.is_a?(Array)
       if type == :int32
         c_data = ::FFI::MemoryPointer.new(:int32, data.count)
-        c_data.put_array_of_int32(0, data)
+        c_data.write_array_of_int32(data)
         check_result FFI.LGBM_DatasetSetField(handle_pointer, field_name, c_data, data.count, 2)
       else
         c_data = ::FFI::MemoryPointer.new(:float, data.count)
-        c_data.put_array_of_float(0, data)
+        c_data.write_array_of_float(data)
         check_result FFI.LGBM_DatasetSetField(handle_pointer, field_name, c_data, data.count, 0)
       end
     end

data/lib/lightgbm/version.rb

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

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: lightgbm
 version: !ruby/object:Gem::Version
-  version: 0.1.6
+  version: 0.1.7
 platform: ruby
 authors:
 - Andrew Kane
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2019-09-30 00:00:00.000000000 Z
+date: 2019-12-06 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: ffi