diff --git a/src/decomposition/pca.rs b/src/decomposition/pca.rs index 7bc53e6..a2c7126 100644 --- a/src/decomposition/pca.rs +++ b/src/decomposition/pca.rs @@ -323,7 +323,7 @@ mod tests { .approximate_eq(&expected_eigenvectors.abs(), 1e-4)); for i in 0..pca.eigenvalues.len() { - assert_eq!(pca.eigenvalues[i].abs(), expected_eigenvalues[i].abs()); + assert!((pca.eigenvalues[i].abs() - expected_eigenvalues[i].abs()).abs() < 1e-8); } let us_arrests_t = pca.transform(&us_arrests); @@ -438,7 +438,7 @@ mod tests { .approximate_eq(&expected_eigenvectors.abs(), 1e-4)); for i in 0..pca.eigenvalues.len() { - assert_eq!(pca.eigenvalues[i].abs(), expected_eigenvalues[i].abs()); + assert!((pca.eigenvalues[i].abs() - expected_eigenvalues[i].abs()).abs() < 1e-8); } let us_arrests_t = pca.transform(&us_arrests); diff --git a/src/linear/logistic_regression.rs b/src/linear/logistic_regression.rs index 8c427ac..cd67cc7 100644 --- a/src/linear/logistic_regression.rs +++ b/src/linear/logistic_regression.rs @@ -264,8 +264,7 @@ impl> LogisticRegression { #[cfg(test)] mod tests { use super::*; - use crate::linalg::naive::dense_matrix::*; - use crate::metrics::*; + use crate::linalg::naive::dense_matrix::*; use ndarray::{arr1, arr2, Array1}; #[test]