xgb 0.10.0 → 0.11.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: c717478c585e099431318c391c7c054b58c844f5ddb19dd5d23f921acbea5b26
-  data.tar.gz: dd221a13627c57135ef2e6e87f267889a76692f1417014e5728b5c279b44a2d5
+  metadata.gz: a7428aac799cd2099ef2bc224c29436b63193b7e419950660f8f6580bd232805
+  data.tar.gz: f266b4e62bafccc3f9079a946defee85d775acc16f2b242b8402c9facc29a5ed
 SHA512:
-  metadata.gz: 4430473a5999cde2447f0782be837f1fb06db44eb3e823621ae94db96b356b40e04a772fa211819ed5915324a10b54ccb0b595d7368aaa22d31f6be4eb13fd0d
-  data.tar.gz: eccc8f38705f82f2aa963c672df5515aa345926af91f9f94beab5fd01406a8d5ad84807d3609dec321952d28d8b2da575b880220ae8375d4296a2c34bf6b0b97
+  metadata.gz: 57da6c5369b485d77c30651b5b33e241b5bc7d7fefb4a4271d5a819ced99e0cf9d3c7a10a0905da8fb18e317fb5021d12a6bb60af954562a22336d008ac78cf7
+  data.tar.gz: c4b17d123505772002444cd12372b2cae5e1205a8c24a2b08fdb815bd60733d3bce57af9592c636369ac7bb10a2fece82f58a070b9a6950447843f5322cae327

data/CHANGELOG.md

@@ -1,3 +1,12 @@
+## 0.11.1 (2026-02-14)
+
+- Updated XGBoost to 3.2.0
+
+## 0.11.0 (2025-10-18)
+
+- Updated XGBoost to 3.1.0
+- Dropped support for Ruby < 3.2
+
 ## 0.10.0 (2025-03-15)
 
 - Updated XGBoost to 3.0.0

data/README.md

@@ -45,13 +45,13 @@ booster.predict(dtest)
 Save the model to a file
 
 ```ruby
-booster.save_model("my.model")
+booster.save_model("model.json")
 ```
 
 Load the model from a file
 
 ```ruby
-booster = XGBoost::Booster.new(model_file: "my.model")
+booster = XGBoost::Booster.new(model_file: "model.json")
 ```
 
 Get the importance of features
@@ -108,13 +108,13 @@ model.predict(x)
 Save the model to a file
 
 ```ruby
-model.save_model("my.model")
+model.save_model("model.json")
 ```
 
 Load the model from a file
 
 ```ruby
-model.load_model("my.model")
+model.load_model("model.json")
 ```
 
 Get the importance of features

data/lib/xgboost/booster.rb

@@ -183,7 +183,7 @@ module XGBoost
 
       names = feature_names || []
       fnames = array_of_pointers(names.map { |fname| string_pointer(fname) })
-      ftypes = array_of_pointers(feature_types || Array.new(names.size, string_pointer("float")))
+      ftypes = array_of_pointers(feature_types&.map { |v| string_pointer(v) } || Array.new(names.size, string_pointer("float")))
 
       check_call FFI.XGBoosterDumpModelExWithFeatures(handle, names.size, fnames, ftypes, with_stats ? 1 : 0, dump_format, out_len, out_result)
 

data/lib/xgboost/callback_container.rb

@@ -110,7 +110,6 @@ module XGBoost
 
     def aggcv(rlist)
       cvmap = {}
-      idx = rlist[0].split[0]
       rlist.each do |line|
         arr = line.split
         arr[1..].each_with_index do |it, metric_idx|
@@ -118,7 +117,6 @@ module XGBoost
           (cvmap[[metric_idx, k]] ||= []) << v.to_f
         end
       end
-      msg = idx
       results = []
       cvmap.sort { |x| x[0][0] }.each do |(_, name), s|
         mean = mean(s)

data/lib/xgboost/classifier.rb

@@ -1,6 +1,6 @@
 module XGBoost
   class Classifier < Model
-    def initialize(n_estimators: 100, objective: "binary:logistic", importance_type: "gain", **options)
+    def initialize(objective: "binary:logistic", **options)
       super
     end
 

data/lib/xgboost/dmatrix.rb

@@ -34,6 +34,17 @@ module XGBoost
       elsif rover?(data)
         nrow, ncol = data.shape
         feature_names = data.keys
+        feature_types =
+          data.types.map do |_, v|
+            v = v.to_s
+            if v.start_with?("int") || v.start_with?("uint")
+              "int"
+            elsif v.start_with?("float")
+              "float"
+            else
+              raise Error, "Unknown feature type: #{v}"
+            end
+          end
         data = data.to_numo
       else
         nrow = data.count
@@ -56,7 +67,7 @@ module XGBoost
       check_call FFI.XGDMatrixCreateFromMat(c_data, nrow, ncol, missing, out)
       @handle = ::FFI::AutoPointer.new(out.read_pointer, FFI.method(:XGDMatrixFree))
 
-      self.feature_names = feature_names || ncol.times.map { |i| "f#{i}" }
+      self.feature_names = feature_names if feature_names
       self.feature_types = feature_types if feature_types
 
       self.label = label if label
@@ -76,9 +87,15 @@ module XGBoost
     end
 
     def group=(group)
-      c_data = ::FFI::MemoryPointer.new(:int, group.size)
-      c_data.write_array_of_int(group)
-      check_call FFI.XGDMatrixSetUIntInfo(handle, "group", c_data, group.size)
+      c_data = ::FFI::MemoryPointer.new(:uint32, group.size)
+      c_data.write_array_of_uint32(group)
+      interface = {
+        shape: [group.length],
+        typestr: "|u4",
+        data: [c_data.address, false],
+        version: 3
+      }
+      check_call FFI.XGDMatrixSetInfoFromInterface(handle, "group", JSON.generate(interface))
     end
 
     def label
@@ -89,6 +106,10 @@ module XGBoost
       float_info("weight")
     end
 
+    def group
+      uint_info("group_ptr")
+    end
+
     def num_row
       out = ::FFI::MemoryPointer.new(:uint64)
       check_call FFI.XGDMatrixNumRow(handle, out)
@@ -234,7 +255,15 @@ module XGBoost
       out_len = ::FFI::MemoryPointer.new(:uint64)
       out_dptr = ::FFI::MemoryPointer.new(:float, num_row)
       check_call FFI.XGDMatrixGetFloatInfo(handle, field, out_len, out_dptr)
-      out_dptr.read_pointer.read_array_of_float(num_row)
+      out_dptr.read_pointer.null? ? nil : out_dptr.read_pointer.read_array_of_float(num_row)
+    end
+
+    def uint_info(field)
+      num_row ||= num_row()
+      out_len = ::FFI::MemoryPointer.new(:uint64)
+      out_dptr = ::FFI::MemoryPointer.new(:uint32, num_row)
+      check_call FFI.XGDMatrixGetUIntInfo(handle, field, out_len, out_dptr)
+      out_dptr.read_pointer.null? ? nil : out_dptr.read_pointer.read_array_of_uint32(num_row)
     end
 
     def validate_feature_info(feature_info, n_features, is_column_split, name)

data/lib/xgboost/ffi.rb

@@ -21,7 +21,7 @@ module XGBoost
 
     # dmatrix
     attach_function :XGDMatrixCreateFromMat, %i[pointer uint64 uint64 float pointer], :int
-    attach_function :XGDMatrixSetUIntInfo, %i[pointer string pointer uint64], :int
+    attach_function :XGDMatrixSetInfoFromInterface, %i[pointer string string], :int
     attach_function :XGDMatrixSetStrFeatureInfo, %i[pointer string pointer uint64], :int
     attach_function :XGDMatrixGetStrFeatureInfo, %i[pointer string pointer pointer], :int
     attach_function :XGDMatrixNumRow, %i[pointer pointer], :int
@@ -33,6 +33,7 @@ module XGBoost
     attach_function :XGDMatrixSaveBinary, %i[pointer string int], :int
     attach_function :XGDMatrixSetFloatInfo, %i[pointer string pointer uint64], :int
     attach_function :XGDMatrixGetFloatInfo, %i[pointer string pointer pointer], :int
+    attach_function :XGDMatrixGetUIntInfo, %i[pointer string pointer pointer], :int
 
     # booster
     attach_function :XGBoosterCreate, %i[pointer int pointer], :int

data/lib/xgboost/model.rb

@@ -19,12 +19,13 @@ module XGBoost
     end
 
     def load_model(fname)
-      @booster = Booster.new(params: @params, model_file: fname)
+      @booster = Booster.new(model_file: fname)
     end
 
     def feature_importances
       score = @booster.score(importance_type: @importance_type)
-      scores = @booster.feature_names.map { |k| score[k] || 0.0 }
+      feature_names = @booster.feature_names || @booster.num_features.times.map { |i| "f#{i}" }
+      scores = feature_names.map { |k| score[k] || 0.0 }
       total = scores.sum.to_f
       scores.map { |s| s / total }
     end

data/lib/xgboost/regressor.rb

@@ -1,6 +1,6 @@
 module XGBoost
   class Regressor < Model
-    def initialize(n_estimators: 100, objective: "reg:squarederror", importance_type: "gain", **options)
+    def initialize(objective: "reg:squarederror", **options)
       super
     end
 

data/lib/xgboost/utils.rb

@@ -23,11 +23,9 @@ module XGBoost
     end
 
     def from_cstr_to_rbstr(data, length)
-      res = []
-      length.read_uint64.times do |i|
-        res << data.read_pointer[i * ::FFI::Pointer.size].read_pointer.read_string.force_encoding(Encoding::UTF_8)
+      data.read_pointer.read_array_of_pointer(length.read_uint64).map do |ptr|
+        ptr.read_string.force_encoding(Encoding::UTF_8)
       end
-      res
     end
   end
 end

data/lib/xgboost/version.rb

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

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification
 name: xgb
 version: !ruby/object:Gem::Version
-  version: 0.10.0
+  version: 0.11.1
 platform: ruby
 authors:
 - Andrew Kane
 bindir: bin
 cert_chain: []
-date: 2025-03-15 00:00:00.000000000 Z
+date: 1980-01-02 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: ffi
@@ -78,14 +78,14 @@ required_ruby_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
-      version: '3.1'
+      version: '3.2'
 required_rubygems_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.6.2
+rubygems_version: 4.0.3
 specification_version: 4
 summary: High performance gradient boosting for Ruby
 test_files: []