File tree 1 file changed +6
-0
lines changed
Filter options
pgml-extension/src/bindings
1 file changed +6
-0
lines changed
Original file line number Diff line number Diff line change @@ -129,6 +129,7 @@ fn get_tree_params(hyperparams: &Hyperparams) -> tree::TreeBoosterParameters {
129
129
"max_bin" => params. max_bin ( value. as_u64 ( ) . unwrap ( ) as u32 ) ,
130
130
"booster" | "n_estimators" | "boost_rounds" => & mut params, // Valid but not relevant to this section
131
131
"nthread" => & mut params,
132
+ "random_state" => & mut params,
132
133
_ => panic ! ( "Unknown hyperparameter {:?}: {:?}" , key, value) ,
133
134
} ;
134
135
}
@@ -161,8 +162,13 @@ fn fit(
161
162
// specify datasets to evaluate against during training
162
163
let evaluation_sets = & [ ( & dtrain, "train" ) , ( & dtest, "test" ) ] ;
163
164
165
+ let seed = match hyperparams. get ( "random_state" ) {
166
+ Some ( value) => value. as_u64 ( ) . unwrap ( ) ,
167
+ None => 0
168
+ } ;
164
169
let learning_params = learning:: LearningTaskParametersBuilder :: default ( )
165
170
. objective ( objective)
171
+ . seed ( seed)
166
172
. build ( )
167
173
. unwrap ( ) ;
168
174
You can’t perform that action at this time.
0 commit comments