Movatterモバイル変換

Skip to content

postgresml/postgresmlPublic

NotificationsYou must be signed in to change notification settings
Fork328
Star6.4k

Commit46f4936

levkk

authored

Fix default XGBoost n_estimators (#338)

1 parent17d1199 commit46f4936Copy full SHA for 46f4936

File tree

3 files changed

+4

-4

lines changed

pgml-extension/pgml_rust/src
- engines
  - xgboost.rs
- lib.rs
- orm
  - estimator.rs

3 files changed

+4

-4

lines changed

`‎pgml-extension/pgml_rust/src/engines/xgboost.rs`

Lines changed: 2 additions & 2 deletions

Original file line number	Diff line number	Diff line change
`@@ -186,8 +186,8 @@ pub fn xgboost_train(`
`186`	`186`	`let params = parameters::TrainingParametersBuilder::default()`
`187`	`187`	`.dtrain(&dtrain)// dataset to train with`
`188`	`188`	`.boost_rounds(match hyperparams.get("n_estimators"){`
`189`		`-Some(value) => value.as_u64().unwrap_or(2)asu32,`
`190`		`-None =>2,`
	`189`	`+Some(value) => value.as_u64().unwrap_or(10)asu32,`
	`190`	`+None =>10,`
`191`	`191`	`})// number of training iterations`
`192`	`192`	`.booster_params(booster_params)// model parameters`
`193`	`193`	`.evaluation_sets(Some(evaluation_sets))// optional datasets to evaluate against in each iteration`

`‎pgml-extension/pgml_rust/src/lib.rs`

Lines changed: 1 addition & 1 deletion

Original file line number	Diff line number	Diff line change
`@@ -3,10 +3,10 @@ extern crate openblas_src;`
`3`	`3`	`externcrate serde;`
`4`	`4`
`5`	`5`	`use once_cell::sync::Lazy;// 1.3.1`
	`6`	`+use parking_lot::Mutex;`
`6`	`7`	`use pgx::*;`
`7`	`8`	`use std::collections::HashMap;`
`8`	`9`	`use std::fs;`
`9`		`-use parking_lot::Mutex;`
`10`	`10`	`use xgboost::{Booster,DMatrix};`
`11`	`11`
`12`	`12`	`pubmod api;`

`‎pgml-extension/pgml_rust/src/orm/estimator.rs`

Lines changed: 1 addition & 1 deletion

Original file line number	Diff line number	Diff line change
`@@ -1,8 +1,8 @@`
	`1`	`+use parking_lot::Mutex;`
`1`	`2`	`use std::collections::HashMap;`
`2`	`3`	`use std::fmt::Debug;`
`3`	`4`	`use std::str::FromStr;`
`4`	`5`	`use std::sync::Arc;`
`5`		`-use parking_lot::Mutex;`
`6`	`6`
`7`	`7`	`use ndarray::{Array1,Array2};`
`8`	`8`	`use once_cell::sync::Lazy;`

0 commit comments

Comments

(0)

[8]ページ先頭

©2009-2025 Movatter.jp