patrick/little_learner
1
0
Fork 0
Code Issues Pull Requests Packages Releases Activity

Fix the run (#23)

Browse Source
This commit is contained in:
Patrick Stevens
2023-05-08 11:55:45 +01:00
committed by GitHub
parent a0da79591a
commit fac93253f2
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
little_learner_app/src/main.rs
View File

@@ -25,7 +25,7 @@ const PLANE_YS: [f64; 6] = [13.99, 15.99, 18.0, 22.4, 30.2, 37.94];
fn main() { fn main() {
let beta = NotNan::new(0.9).expect("not nan"); let beta = NotNan::new(0.9).expect("not nan");
let stabilizer = NotNan::new(0.000_000_01).expect("not nan"); let stabilizer = NotNan::new(0.000_000_01).expect("not nan");
let hyper = hyper::RmsGradientDescent::default(NotNan::new(0.001).expect("not nan"), 3000) let hyper = hyper::RmsGradientDescent::default(NotNan::new(0.01).expect("not nan"), 3000)
.with_stabilizer(stabilizer) .with_stabilizer(stabilizer)
.with_beta(beta); .with_beta(beta);
@@ -62,9 +62,9 @@ fn main() {
let fitted_theta1 = theta1.to_scalar().real_part().into_inner(); let fitted_theta1 = theta1.to_scalar().real_part().into_inner();
assert_eq!( assert_eq!(
fitted_theta0, fitted_theta0,
[3.985_350_099_342_649, 1.974_594_572_821_635_2] [3.974_645_444_172_085, 1.971_454_922_077_495]
); );
assert_eq!(fitted_theta1, 6.164_222_983_181_168); assert_eq!(fitted_theta1, 6.164_579_048_274_036);
} }
#[cfg(test)] #[cfg(test)]