class LightGBM::Model
Attributes
booster[R]
Public Class Methods
new(num_leaves: 31, learning_rate: 0.1, n_estimators: 100, objective: nil, **options)
click to toggle source
# File lib/lightgbm/model.rb, line 5 def initialize(num_leaves: 31, learning_rate: 0.1, n_estimators: 100, objective: nil, **options) @params = { num_leaves: num_leaves, learning_rate: learning_rate }.merge(options) @params[:objective] = objective if objective @n_estimators = n_estimators end
Public Instance Methods
best_iteration()
click to toggle source
# File lib/lightgbm/model.rb, line 22 def best_iteration @booster.best_iteration end
feature_importances()
click to toggle source
# File lib/lightgbm/model.rb, line 26 def feature_importances @booster.feature_importance end
load_model(fname)
click to toggle source
# File lib/lightgbm/model.rb, line 18 def load_model(fname) @booster = Booster.new(params: @params, model_file: fname) end
save_model(fname)
click to toggle source
# File lib/lightgbm/model.rb, line 14 def save_model(fname) @booster.save_model(fname) end