@@ -41,7 +41,7 @@ def dump_model(num_iteration: nil, start_iteration: 0)
4141 out_str = ::FFI ::MemoryPointer . new ( :char , buffer_len )
4242 feature_importance_type = 0 # TODO add option
4343 check_result FFI . LGBM_BoosterDumpModel ( handle_pointer , start_iteration , num_iteration , feature_importance_type , buffer_len , out_len , out_str )
44- actual_len = read_int64 ( out_len )
44+ actual_len = out_len . read_int64
4545 if actual_len > buffer_len
4646 out_str = ::FFI ::MemoryPointer . new ( :char , actual_len )
4747 check_result FFI . LGBM_BoosterDumpModel ( handle_pointer , start_iteration , num_iteration , feature_importance_type , actual_len , out_len , out_str )
@@ -110,7 +110,7 @@ def model_to_string(num_iteration: nil, start_iteration: 0)
110110 out_str = ::FFI ::MemoryPointer . new ( :char , buffer_len )
111111 feature_importance_type = 0 # TODO add option
112112 check_result FFI . LGBM_BoosterSaveModelToString ( handle_pointer , start_iteration , num_iteration , feature_importance_type , buffer_len , out_len , out_str )
113- actual_len = read_int64 ( out_len )
113+ actual_len = out_len . read_int64
114114 if actual_len > buffer_len
115115 out_str = ::FFI ::MemoryPointer . new ( :char , actual_len )
116116 check_result FFI . LGBM_BoosterSaveModelToString ( handle_pointer , start_iteration , num_iteration , feature_importance_type , actual_len , out_len , out_str )
@@ -168,7 +168,7 @@ def predict(input, start_iteration: nil, num_iteration: nil, **params)
168168 out_len = ::FFI ::MemoryPointer . new ( :int64 )
169169 out_result = ::FFI ::MemoryPointer . new ( :double , num_class * input . count )
170170 check_result FFI . LGBM_BoosterPredictForMat ( handle_pointer , data , 1 , input . count , input . first . count , 1 , 0 , start_iteration , num_iteration , params_str ( params ) , out_len , out_result )
171- out = out_result . read_array_of_double ( read_int64 ( out_len ) )
171+ out = out_result . read_array_of_double ( out_len . read_int64 )
172172 out = out . each_slice ( num_class ) . to_a if num_class > 1
173173
174174 singular ? out . first : out
@@ -244,11 +244,6 @@ def num_class
244244 out . read_int
245245 end
246246
247- # read_int64 not available on JRuby
248- def read_int64 ( ptr )
249- ptr . read_array_of_int64 ( 1 ) . first
250- end
251-
252247 def sorted_feature_values ( input_hash )
253248 input_hash . transform_keys ( &:to_s ) . fetch_values ( *cached_feature_name )
254249 end
0 commit comments