magnus167/rustframe
1
0
Fork 0
mirror of https://github.com/Magnus167/rustframe.git synced 2025-08-20 04:00:01 +00:00
Code Issues Packages Projects Releases Wiki Activity

Relax assertion tolerance in KMeans tests to align with algorithm's convergence criteria

Browse Source
This commit is contained in:
Palash Tyagi 2025-07-12 01:47:40 +01:00
parent 12a72317e4
commit c24eb4a08c
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/compute/models/k_means.rs
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_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; 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); // Relax the assertion tolerance to match the algorithm's convergence tolerance
assert!((kmeans_model.centroids[(0, 1)] - expected_centroid_y).abs() < 1e-9); 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] #[test]