fix: broken unit test

This commit is contained in:
Volodymyr Orlov
2020-09-14 16:15:15 -07:00
parent fa02d77ab1
commit 01f536d7f7
+2 -3
View File
@@ -181,6 +181,7 @@ impl<T: RealNumber> RandomForestRegressor<T> {
mod tests { mod tests {
use super::*; use super::*;
use crate::linalg::naive::dense_matrix::DenseMatrix; use crate::linalg::naive::dense_matrix::DenseMatrix;
use crate::metrics::mean_absolute_error;
#[test] #[test]
fn fit_longley() { fn fit_longley() {
@@ -224,9 +225,7 @@ mod tests {
) )
.predict(&x); .predict(&x);
for i in 0..y_hat.len() { assert!(mean_absolute_error(&y, &y_hat) < 1.0);
assert!((y_hat[i] - expected_y[i]).abs() < 1.0);
}
} }
#[test] #[test]