xgb 0.9.0 → 0.10.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: ba232bc8a9c27bc9cfb2b7f87afc581acf9a5464df9fccf1b206d912704771fe
-  data.tar.gz: 608ca80f01e81d32a31b796364de14096c89556543fa2892efd86bbf86fac96a
+  metadata.gz: c717478c585e099431318c391c7c054b58c844f5ddb19dd5d23f921acbea5b26
+  data.tar.gz: dd221a13627c57135ef2e6e87f267889a76692f1417014e5728b5c279b44a2d5
 SHA512:
-  metadata.gz: af38ca270cf3d12a8a7757ff5f32a41f5d68e43f9ad08c585df8efae835d014ddcac51201019aa339143cc081c5c69ef5914d8de06cb4326ccb2864b5b04eba2
-  data.tar.gz: 787bddc1c867d648ffe13c783aede98c62917feca536d1f8fe27e7992e5f2945126ad15b1a4e3ff48cc041723d299cf3cbacdd5557083983575eae22a9583cd8
+  metadata.gz: 4430473a5999cde2447f0782be837f1fb06db44eb3e823621ae94db96b356b40e04a772fa211819ed5915324a10b54ccb0b595d7368aaa22d31f6be4eb13fd0d
+  data.tar.gz: eccc8f38705f82f2aa963c672df5515aa345926af91f9f94beab5fd01406a8d5ad84807d3609dec321952d28d8b2da575b880220ae8375d4296a2c34bf6b0b97

data/CHANGELOG.md

@@ -1,3 +1,7 @@
+## 0.10.0 (2025-03-15)
+
+- Updated XGBoost to 3.0.0
+
 ## 0.9.0 (2024-10-17)
 
 - Updated XGBoost to 2.1.1

data/NOTICE.txt

@@ -1,5 +1,5 @@
-Copyright XGBoost contributors
-Copyright 2019-2024 Andrew Kane
+Copyright 2014-2025 XGBoost contributors
+Copyright 2019-2025 Andrew Kane
 
 Licensed under the Apache License, Version 2.0 (the "License");
 you may not use this file except in compliance with the License.

data/README.md

@@ -126,7 +126,8 @@ model.feature_importances
 Early stopping
 
 ```ruby
-model.fit(x, y, eval_set: [[x_test, y_test]], early_stopping_rounds: 5)
+model = XGBoost::Regressor.new(early_stopping_rounds: 5)
+model.fit(x, y, eval_set: [[x_test, y_test]])
 ```
 
 ## Data

data/lib/xgboost/booster.rb

@@ -47,6 +47,11 @@ module XGBoost
       json_string.read_pointer.read_string(length.read_uint64).force_encoding(Encoding::UTF_8)
     end
 
+    def reset
+      check_call FFI.XGBoosterReset(handle)
+      self
+    end
+
     def attr(key)
       ret = ::FFI::MemoryPointer.new(:pointer)
       success = ::FFI::MemoryPointer.new(:int)

data/lib/xgboost/classifier.rb

@@ -18,7 +18,7 @@ module XGBoost
 
       @booster = XGBoost.train(params, dtrain,
         num_boost_round: @n_estimators,
-        early_stopping_rounds: early_stopping_rounds,
+        early_stopping_rounds: early_stopping_rounds || @early_stopping_rounds,
         verbose_eval: verbose,
         evals: evals
       )

data/lib/xgboost/ffi.rb

@@ -39,6 +39,7 @@ module XGBoost
     attach_function :XGBoosterUpdateOneIter, %i[pointer int pointer], :int
     attach_function :XGBoosterEvalOneIter, %i[pointer int pointer pointer uint64 pointer], :int
     attach_function :XGBoosterFree, %i[pointer], :int
+    attach_function :XGBoosterReset, %i[pointer], :int
     attach_function :XGBoosterBoostedRounds, %i[pointer pointer], :int
     attach_function :XGBoosterSetParam, %i[pointer string string], :int
     attach_function :XGBoosterGetNumFeature, %i[pointer pointer], :int

data/lib/xgboost/model.rb

@@ -2,10 +2,11 @@ module XGBoost
   class Model
     attr_reader :booster
 
-    def initialize(n_estimators: 100, importance_type: "gain", **options)
+    def initialize(n_estimators: 100, importance_type: "gain", early_stopping_rounds: nil, **options)
       @params = options
       @n_estimators = n_estimators
       @importance_type = importance_type
+      @early_stopping_rounds = early_stopping_rounds
     end
 
     def predict(data)

data/lib/xgboost/regressor.rb

@@ -10,7 +10,7 @@ module XGBoost
 
       @booster = XGBoost.train(@params, dtrain,
         num_boost_round: @n_estimators,
-        early_stopping_rounds: early_stopping_rounds,
+        early_stopping_rounds: early_stopping_rounds || @early_stopping_rounds,
         verbose_eval: verbose,
         evals: evals
       )

data/lib/xgboost/version.rb

@@ -1,3 +1,3 @@
 module XGBoost
-  VERSION = "0.9.0"
+  VERSION = "0.10.0"
 end

data/lib/xgboost.rb

@@ -91,7 +91,7 @@ module XGBoost
         evals_result.merge!(cb_container.history)
       end
 
-      bst
+      bst.reset
     end
 
     def cv(

metadata

@@ -1,14 +1,13 @@
 --- !ruby/object:Gem::Specification
 name: xgb
 version: !ruby/object:Gem::Version
-  version: 0.9.0
+  version: 0.10.0
 platform: ruby
 authors:
 - Andrew Kane
-autorequire:
 bindir: bin
 cert_chain: []
-date: 2024-10-17 00:00:00.000000000 Z
+date: 2025-03-15 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: ffi
@@ -24,7 +23,6 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '0'
-description:
 email: andrew@ankane.org
 executables: []
 extensions: []
@@ -73,7 +71,6 @@ homepage: https://github.com/ankane/xgboost-ruby
 licenses:
 - Apache-2.0
 metadata: {}
-post_install_message:
 rdoc_options: []
 require_paths:
 - lib
@@ -88,8 +85,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.5.16
-signing_key:
+rubygems_version: 3.6.2
 specification_version: 4
 summary: High performance gradient boosting for Ruby
 test_files: []