lib/xgboost/model.rb in xgb-0.2.1 vs lib/xgboost/model.rb in xgb-0.3.0

@@ -1,14 +1,10 @@
 module XGBoost
   class Model
     attr_reader :booster
 
-    def initialize(max_depth: 3, learning_rate: 0.1, n_estimators: 100, objective: nil, importance_type: "gain", **options)
-      @params = {
-        max_depth: max_depth,
-        objective: objective,
-        learning_rate: learning_rate
-      }.merge(options)
+    def initialize(n_estimators: 100, importance_type: "gain", **options)
+      @params = options
       @n_estimators = n_estimators
       @importance_type = importance_type
     end
 
     def predict(data)