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

Refactor test for KMeans empty cluster reinitialization to use distinct data points and remove redundant assertion

Browse Source
This commit is contained in:
Palash Tyagi 2025-07-13 01:49:15 +01:00
parent 7c7c8c2a16
commit 1bdcf1b113
1 changed files with 6 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
src/compute/models/k_means.rs
View File

@ -328,13 +328,12 @@ mod tests {
// Two distinct groups of points, but we ask for 3 clusters.
// This should cause one cluster to be empty and re-initialized.
let data = vec![
1.0, 1.0,
1.1, 1.1,
1.2, 1.2,
// Large gap to ensure distinct clusters
100.0, 100.0,
100.1, 100.1,
100.2, 100.2,
0.0, 0.0, // Single point cluster
10.0, 10.0,
10.1, 10.1,
10.2, 10.2,
10.3, 10.3,
10.4, 10.4,
];
let x = FloatMatrix::from_rows_vec(data, 6, 2);
let k = 3; // Request 3 clusters for 2 natural groups
@ -362,10 +361,6 @@ mod tests {
counts[label] += 1;
}
// The crucial assertion: After re-initialization, no cluster should remain empty.
// This verifies that the "furthest point" logic successfully re-assigned a point
// to the previously empty cluster.
assert!(counts.iter().all(|&c| c > 0));
// The crucial assertion: After re-initialization, no cluster should remain empty.
// This verifies that the "furthest point" logic successfully re-assigned a point