Improved code

ankane · ankane · commit 9c6b2627e564 · 2025-10-18T13:23:38.000-07:00
diff --git a/lib/xgboost/dmatrix.rb b/lib/xgboost/dmatrix.rb
@@ -76,8 +76,8 @@ def weight=(weight)
     end
 
     def group=(group)
-      c_data = ::FFI::MemoryPointer.new(:uint, group.size)
-      c_data.write_array_of_uint(group)
+      c_data = ::FFI::MemoryPointer.new(:uint32, group.size)
+      c_data.write_array_of_uint32(group)
       interface = {
         shape: [group.length],
         typestr: "|u4",
@@ -250,9 +250,9 @@ def float_info(field)
     def uint_info(field)
       num_row ||= num_row()
       out_len = ::FFI::MemoryPointer.new(:uint64)
-      out_dptr = ::FFI::MemoryPointer.new(:uint, num_row)
+      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_uint(num_row)
+      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)
