lib/xgboost/ranker.rb in xgb-0.7.3 vs lib/xgboost/ranker.rb in xgb-0.8.0
- old
+ new
@@ -1,7 +1,7 @@
module XGBoost
class Ranker < Model
- def initialize(n_estimators: 100, objective: "rank:pairwise", importance_type: "gain", **options)
+ def initialize(objective: "rank:ndcg", **options)
super
end
def fit(x, y, group)
dtrain = DMatrix.new(x, label: y)