Skip to content

Commit 0de57e5

Browse files
committed
Removed unneeded code [skip ci]
1 parent a030213 commit 0de57e5

File tree

1 file changed

+3
-8
lines changed

1 file changed

+3
-8
lines changed

lib/lightgbm/booster.rb

Lines changed: 3 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)