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 |