fix(py): n_cluter & max_iter can be 1

Signed-off-by: Keming <[email protected]>
kemingy · Dec 6, 2024 · 1b15e46 · 1b15e46
1 parent c599799
commit 1b15e46
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/kmeans.rs b/src/kmeans.rs
@@ -187,11 +187,11 @@ impl KMeans {
         distance: Distance,
         use_residual: bool,
     ) -> Self {
-        if n_cluster <= 1 {
-            panic!("n_cluster must be greater than 1");
+        if n_cluster < 1 {
+            panic!("n_cluster must be greater than 0");
         }
-        if max_iter <= 1 {
-            panic!("max_iter must be greater than 1");
+        if max_iter < 1 {
+            panic!("max_iter must be greater than 0");
         }
         if tolerance <= 0.0 {
             panic!("tolerance must be greater than 0.0");