lightgbm 0.1.8 → 0.1.9

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: d3b86686bc4575d069e469fb9c8911e93c7dcfb7622a415dc3e7ebee301947c3
-  data.tar.gz: a186cca11d4838fd13573b14ea154fff174804b31ae45236130c30a24f55ff7b
+  metadata.gz: a4aac9eac1ab0dadbf31d1e1fc2714e75a8c37075538aef5c53b42df1c34f658
+  data.tar.gz: ca3a1043c55184992b3fac611963062d01747449f6170719ef1f299d4f0474c9
 SHA512:
-  metadata.gz: 32ef8f452075bcf4441b8e0fcf68e63332c42da61445301ecd0a85b51b2af1b7a730da0810e1b182ff5fdfe184884242a5489469d16e0e9a717d19bb145eb095
-  data.tar.gz: 41b76c5ac174b75ce4e4f21477e4d32a9607b550e70448dde2797a765813d08a6c75de2f7f6d04ea79b4e9fa8afd768d097be516781e27e4eee4f982af364dc0
+  metadata.gz: c2f14ccc3b40690060d2ee533cfe46e137f40e91520aab3eb188a8a03a697a6956853b717e743cd357d0a059836fc32e9e0aa0fbe5a7dd4263b1ff3e94e79601
+  data.tar.gz: 64abdb43f4c45222dcbb39ddde1d21350b2b4958439dc84d8f13684f4e9bcd3aa2c0e49aa8fd9ae1dfc09b8bc0ce69592e95aab0542684f3c952e9e67ac4689f

data/CHANGELOG.md

@@ -1,3 +1,8 @@
+## 0.1.9 (2020-06-10)
+
+- Added support for Rover
+- Improved performance of Numo datasets
+
 ## 0.1.8 (2020-05-09)
 
 - Improved error message when OpenMP not found on Mac

data/README.md

@@ -129,16 +129,22 @@ Data can be an array of arrays
 [[1, 2, 3], [4, 5, 6]]
 ```
 
-Or a Daru data frame
+Or a Numo NArray
 
 ```ruby
-Daru::DataFrame.from_csv("houses.csv")
+Numo::NArray.cast([[1, 2, 3], [4, 5, 6]])
 ```
 
-Or a Numo NArray
+Or a Rover data frame
+
+```ruby
+Rover.read_csv("houses.csv")
+```
+
+Or a Daru data frame
 
 ```ruby
-Numo::DFloat.new(3, 2).seq
+Daru::DataFrame.from_csv("houses.csv")
 ```
 
 ## Helpful Resources

data/lib/lightgbm/dataset.rb

@@ -130,18 +130,23 @@ module LightGBM
         elsif daru?(data)
           nrow, ncol = data.shape
           flat_data = data.map_rows(&:to_a).flatten
-        elsif narray?(data)
+        elsif numo?(data) || rover?(data)
+          data = data.to_numo if rover?(data)
           nrow, ncol = data.shape
-          flat_data = data.flatten.to_a
         else
           nrow = data.count
           ncol = data.first.count
           flat_data = data.flatten
         end
 
-        handle_missing(flat_data)
         c_data = ::FFI::MemoryPointer.new(:double, nrow * ncol)
-        c_data.write_array_of_double(flat_data)
+        if numo?(data)
+          c_data.write_bytes(data.cast_to(Numo::DFloat).to_string)
+        else
+          handle_missing(flat_data)
+          c_data.write_array_of_double(flat_data)
+        end
+
         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

data/lib/lightgbm/utils.rb

@@ -37,8 +37,12 @@ module LightGBM
       defined?(Daru::DataFrame) && data.is_a?(Daru::DataFrame)
     end
 
-    def narray?(data)
+    def numo?(data)
       defined?(Numo::NArray) && data.is_a?(Numo::NArray)
     end
+
+    def rover?(data)
+      defined?(Rover::DataFrame) && data.is_a?(Rover::DataFrame)
+    end
   end
 end

data/lib/lightgbm/version.rb

@@ -1,3 +1,3 @@
 module LightGBM
-  VERSION = "0.1.8"
+  VERSION = "0.1.9"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: lightgbm
 version: !ruby/object:Gem::Version
-  version: 0.1.8
+  version: 0.1.9
 platform: ruby
 authors:
 - Andrew Kane
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2020-05-09 00:00:00.000000000 Z
+date: 2020-06-11 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: ffi