Merge pull request #383 from CoRiis/patch-2

alejandroschuler · web-flow · commit c143d4a07a47 · 2025-03-21T10:50:06.000-07:00
Update parameters in `get_params`
diff --git a/ngboost/ngboost.py b/ngboost/ngboost.py
@@ -512,6 +512,8 @@ def get_params(self, deep=True):
             "col_sample": self.col_sample,
             "verbose": self.verbose,
             "random_state": self.random_state,
+            "validation_fraction": self.validation_fraction,
+            "early_stopping_rounds": self.early_stopping_rounds,
         }
 
         return params

Original file line number	Diff line number	Diff line change
`@@ -512,6 +512,8 @@ def get_params(self, deep=True):`
`512`	`512`	`"col_sample": self.col_sample,`
`513`	`513`	`"verbose": self.verbose,`
`514`	`514`	`"random_state": self.random_state,`
	`515`	`+ "validation_fraction": self.validation_fraction,`
	`516`	`+ "early_stopping_rounds": self.early_stopping_rounds,`
`515`	`517`	`}`
`516`	`518`
`517`	`519`	`return params`