lib/xgboost/regressor.rb in xgb-0.3.1 vs lib/xgboost/regressor.rb in xgb-0.4.0

- old
+ new

@@ -1,7 +1,7 @@ module XGBoost class Regressor < Model - def initialize(max_depth: 3, learning_rate: 0.1, n_estimators: 100, objective: "reg:squarederror", importance_type: "gain", **options) + def initialize(n_estimators: 100, objective: "reg:squarederror", importance_type: "gain", **options) super end def fit(x, y, eval_set: nil, early_stopping_rounds: nil, verbose: true) dtrain = DMatrix.new(x, label: y)