mirror of
https://github.com/Magnus167/rustframe.git
synced 2025-08-21 03:49:59 +00:00
Compare commits
1 Commits
c0f82d5ce8
...
e2bb4a4724
Author | SHA1 | Date | |
---|---|---|---|
![]() |
e2bb4a4724 |
@ -33,22 +33,3 @@ impl LinReg {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
mod tests {
|
|
||||||
|
|
||||||
use super::LinReg;
|
|
||||||
use crate::matrix::{Matrix};
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn test_linreg_fit_predict() {
|
|
||||||
let x = Matrix::from_vec(vec![1.0, 2.0, 3.0, 4.0], 4, 1);
|
|
||||||
let y = Matrix::from_vec(vec![2.0, 3.0, 4.0, 5.0], 4, 1);
|
|
||||||
let mut model = LinReg::new(1);
|
|
||||||
model.fit(&x, &y, 0.01, 10000);
|
|
||||||
let preds = model.predict(&x);
|
|
||||||
assert!((preds[(0, 0)] - 2.0).abs() < 1e-2);
|
|
||||||
assert!((preds[(1, 0)] - 3.0).abs() < 1e-2);
|
|
||||||
assert!((preds[(2, 0)] - 4.0).abs() < 1e-2);
|
|
||||||
assert!((preds[(3, 0)] - 5.0).abs() < 1e-2);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
Loading…
x
Reference in New Issue
Block a user