Class: XGBoost::Ranker

Inherits:
Model
  • Object
show all
Defined in:
lib/xgboost/ranker.rb

Instance Attribute Summary

Attributes inherited from Model

#booster

Instance Method Summary collapse

Methods inherited from Model

#feature_importances, #load_model, #predict, #save_model

Constructor Details

#initialize(n_estimators: 100, objective: "rank:pairwise", importance_type: "gain", **options) ⇒ Ranker

Returns a new instance of Ranker.



3
4
5
 
# File 'lib/xgboost/ranker.rb', line 3

def initialize(n_estimators: 100, objective: "rank:pairwise", importance_type: "gain", **options)
  super
end

Instance Method Details

#fit(x, y, group) ⇒ Object 



7
8
9
10
11
12
 
# File 'lib/xgboost/ranker.rb', line 7

def fit(x, y, group)
  dtrain = DMatrix.new(x, label: y)
  dtrain.group = group
  @booster = XGBoost.train(@params, dtrain, num_boost_round: @n_estimators)
  nil
end