Skip to content

Commit 3ce0fe4

Browse files
authored
Update parameters in get_params
The two parameters `validation_fraction` and `early_stopping_rounds` have been added to the `params` dictionary.
1 parent a6c3c21 commit 3ce0fe4

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

ngboost/ngboost.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -512,6 +512,8 @@ def get_params(self, deep=True):
512512
"col_sample": self.col_sample,
513513
"verbose": self.verbose,
514514
"random_state": self.random_state,
515+
"validation_fraction": self.validation_fraction,
516+
"early_stopping_rounds": self.early_stopping_rounds,
515517
}
516518

517519
return params

0 commit comments

Comments
 (0)