Contents

module XGBoost
  class Ranker < Model
    def initialize(objective: "rank:ndcg", **options)
      super
    end

    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
  end
end

Version data entries

3 entries across 3 versions & 2 rubygems

Version	Path
honzasterba_xgb-0.9.0	lib/xgboost/ranker.rb
xgb-0.9.0	lib/xgboost/ranker.rb
xgb-0.8.0	lib/xgboost/ranker.rb