2 Commits

Author SHA1 Message Date
Palash Tyagi
727ec91a8b Merge c24eb4a08c into 11330e464b 2025-07-12 00:47:45 +00:00
Palash Tyagi
c24eb4a08c Relax assertion tolerance in KMeans tests to align with algorithm's convergence criteria 2025-07-12 01:47:40 +01:00

View File

@@ -265,8 +265,9 @@ mod tests {
let expected_centroid_x = x.column(0).iter().sum::<f64>() / x.rows() as f64;
let expected_centroid_y = x.column(1).iter().sum::<f64>() / x.rows() as f64;
assert!((kmeans_model.centroids[(0, 0)] - expected_centroid_x).abs() < 1e-9);
assert!((kmeans_model.centroids[(0, 1)] - expected_centroid_y).abs() < 1e-9);
// Relax the assertion tolerance to match the algorithm's convergence tolerance
assert!((kmeans_model.centroids[(0, 0)] - expected_centroid_x).abs() < 1e-6);
assert!((kmeans_model.centroids[(0, 1)] - expected_centroid_y).abs() < 1e-6);
}
#[test]