lightgbm 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: c7222cd44123f8c35d26b91dbf0288e3d0f7a87e3204a6a501f7590648896d2f
+  data.tar.gz: 2a288731615d9042b0b60184c534bd00d3d64d1d8be1b8dbc2aecb782fd3332b
+SHA512:
+  metadata.gz: 290d729343c88d2054082692cff9dc948332933439b11915bd507338dc28354fec936b7eded198c8a6bd18d9821ff32d1dccc146c71db7567ddc6037838c0e4b
+  data.tar.gz: 24cd193a72188d43b71cc223d45cc943d710149d42b7ee0fb11eeab8fdf0a9e006ae021313d0881d91cebfbc510a08864721af42f4976e5cd0663ca1a6534f39

data/CHANGELOG.md

@@ -0,0 +1,3 @@
+## 0.1.0
+
+- First release

data/README.md

@@ -0,0 +1,42 @@
+# LightGBM
+
+LightGBM for Ruby
+
+## Installation
+
+First, [install LightGBM](https://lightgbm.readthedocs.io/en/latest/Installation-Guide.html).
+
+Add this line to your application’s Gemfile:
+
+```ruby
+gem 'lightgbm'
+```
+
+Load a model
+
+```ruby
+booster = LightGBM::Booster.new(model_file: "model.txt")
+```
+
+Predict
+
+```ruby
+booster.predict([[1, 2], [3, 4]])
+```
+
+## Credits
+
+Thanks to the [xgboost](https://github.com/PairOnAir/xgboost-ruby) gem for serving as an initial reference.
+
+## History
+
+View the [changelog](https://github.com/ankane/lightgbm/blob/master/CHANGELOG.md)
+
+## Contributing
+
+Everyone is encouraged to help improve this project. Here are a few ways you can help:
+
+- [Report bugs](https://github.com/ankane/lightgbm/issues)
+- Fix bugs and [submit pull requests](https://github.com/ankane/lightgbm/pulls)
+- Write, clarify, or fix documentation
+- Suggest or add new features

data/lib/lightgbm.rb

@@ -0,0 +1,7 @@
+# dependencies
+require "ffi"
+
+# modules
+require "lightgbm/booster"
+require "lightgbm/ffi"
+require "lightgbm/version"

data/lib/lightgbm/booster.rb

@@ -0,0 +1,46 @@
+module LightGBM
+  class Booster
+    def initialize(model_file:)
+      @handle = ::FFI::MemoryPointer.new(:pointer)
+      if model_file
+        out_num_iterations = ::FFI::MemoryPointer.new(:int)
+        check_result FFI.LGBM_BoosterCreateFromModelfile(model_file, out_num_iterations, @handle)
+      end
+      ObjectSpace.define_finalizer(self, self.class.finalize(handle_pointer))
+    end
+
+    def self.finalize(pointer)
+      -> { FFI.LGBM_BoosterFree(pointer) }
+    end
+
+    def predict(input)
+      raise TypeError unless input.is_a?(Array)
+
+      singular = input.first.is_a?(Array)
+      input = [input] unless singular
+
+      data = ::FFI::MemoryPointer.new(:float, input.count * input.first.count)
+      data.put_array_of_float(0, input.flatten)
+
+      out_len = ::FFI::MemoryPointer.new(:int64)
+      out_result = ::FFI::MemoryPointer.new(:double, input.count)
+      parameter = ""
+      check_result FFI.LGBM_BoosterPredictForMat(handle_pointer, data, 0, input.count, input.first.count, 1, 0, 0, parameter, out_len, out_result)
+      out = out_result.read_array_of_double(out_len.read_int64)
+
+      singular ? out : out.first
+    end
+
+    private
+
+    def check_result(err)
+      if err != 0
+        raise FFI.LGBM_GetLastError
+      end
+    end
+
+    def handle_pointer
+      @handle.read_pointer
+    end
+  end
+end

data/lib/lightgbm/ffi.rb

@@ -0,0 +1,13 @@
+module LightGBM
+  module FFI
+    extend ::FFI::Library
+    ffi_lib ["lightgbm", "lib_lightgbm.so"]
+
+    # https://github.com/microsoft/LightGBM/blob/master/include/LightGBM/c_api.h
+    attach_function :LGBM_GetLastError, %i[], :string
+    attach_function :LGBM_BoosterCreate, %i[pointer string pointer], :int
+    attach_function :LGBM_BoosterCreateFromModelfile, %i[string pointer pointer], :int
+    attach_function :LGBM_BoosterFree, %i[pointer], :int
+    attach_function :LGBM_BoosterPredictForMat, %i[pointer pointer int int32 int32 int int int string pointer pointer], :int
+  end
+end

data/lib/lightgbm/version.rb

@@ -0,0 +1,3 @@
+module LightGBM
+  VERSION = "0.1.0"
+end

metadata

@@ -0,0 +1,104 @@
+--- !ruby/object:Gem::Specification
+name: lightgbm
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Andrew Kane
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2019-08-14 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: ffi
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: bundler
+  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'
+- !ruby/object:Gem::Dependency
+  name: rake
+  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'
+- !ruby/object:Gem::Dependency
+  name: minitest
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '5'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '5'
+description: 
+email: andrew@chartkick.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- CHANGELOG.md
+- README.md
+- lib/lightgbm.rb
+- lib/lightgbm/booster.rb
+- lib/lightgbm/ffi.rb
+- lib/lightgbm/version.rb
+homepage: https://github.com/ankane/lightgbm
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '2.4'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.0.4
+signing_key: 
+specification_version: 4
+summary: LightGBM for Ruby
+test_files: []