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