lib/xgboost/classifier.rb in xgb-0.3.1 vs lib/xgboost/classifier.rb in xgb-0.4.0
- old
+ new
@@ -1,7 +1,7 @@
module XGBoost
class Classifier < Model
- def initialize(max_depth: 3, learning_rate: 0.1, n_estimators: 100, objective: "binary:logistic", importance_type: "gain", **options)
+ def initialize(n_estimators: 100, objective: "binary:logistic", importance_type: "gain", **options)
super
end
def fit(x, y, eval_set: nil, early_stopping_rounds: nil, verbose: true)
n_classes = y.uniq.size