We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see ourdocumentation.
There was an error while loading.Please reload this page.
1 parent1d513fd commit484b2fcCopy full SHA for 484b2fc
pgml-extension/src/bindings/xgboost.rs
@@ -129,6 +129,7 @@ fn get_tree_params(hyperparams: &Hyperparams) -> tree::TreeBoosterParameters {
129
"max_bin" => params.max_bin(value.as_u64().unwrap()asu32),
130
"booster" |"n_estimators" |"boost_rounds" =>&mut params,// Valid but not relevant to this section
131
"nthread" =>&mut params,
132
+"random_state" =>&mut params,
133
_ =>panic!("Unknown hyperparameter {:?}: {:?}", key, value),
134
};
135
}
@@ -161,8 +162,13 @@ fn fit(
161
162
// specify datasets to evaluate against during training
163
let evaluation_sets =&[(&dtrain,"train"),(&dtest,"test")];
164
165
+let seed =match hyperparams.get("random_state"){
166
+Some(value) => value.as_u64().unwrap(),
167
+None =>0
168
+};
169
let learning_params = learning::LearningTaskParametersBuilder::default()
170
.objective(objective)
171
+.seed(seed)
172
.build()
173
.unwrap();
174