Method: LightGBM::Booster#save_model

Defined in:
lib/lightgbm/booster.rb

#save_model(filename, num_iteration: nil, start_iteration: 0) ⇒ Object



149
150
151
152
153
154
# File 'lib/lightgbm/booster.rb', line 149

def save_model(filename, num_iteration: nil, start_iteration: 0)
  num_iteration ||= best_iteration
  feature_importance_type = 0 # TODO add
  check_result FFI.LGBM_BoosterSaveModel(handle_pointer, start_iteration, num_iteration, feature_importance_type, filename)
  self # consistent with Python API
end