Check cluster centers size in copy constructor of inlined kmeans (net…

…data#19356) Co-authored-by: Costa Tsaousis <costa@netdata.cloud>
webfutureiorepo · Jan 14, 2025 · 592ee98 · 592ee98
1 parent 4f072e6
commit 592ee98
Showing 1 changed file with 6 additions and 8 deletions.
diff --git a/src/ml/ml_kmeans.h b/src/ml/ml_kmeans.h
@@ -37,12 +37,11 @@ struct ml_kmeans_inlined_t {
 
     explicit ml_kmeans_inlined_t(const ml_kmeans_t &km)
     {
-        if (km.cluster_centers.size() != 2) {
-            throw std::runtime_error("ml_kmeans_t must have exactly 2 cluster centers");
+        if (km.cluster_centers.size() == 2) {
+            cluster_centers[0] = km.cluster_centers[0];
+            cluster_centers[1] = km.cluster_centers[1];
         }
 
-        cluster_centers[0] = km.cluster_centers[0];
-        cluster_centers[1] = km.cluster_centers[1];
         min_dist = km.min_dist;
         max_dist = km.max_dist;
         after = km.after;
@@ -51,11 +50,10 @@ struct ml_kmeans_inlined_t {
 
     ml_kmeans_inlined_t &operator=(const ml_kmeans_t &km)
     {
-        if (km.cluster_centers.size() != 2) {
-            throw std::runtime_error("ml_kmeans_t must have exactly 2 cluster centers");
+        if (km.cluster_centers.size() == 2) {
+            cluster_centers[0] = km.cluster_centers[0];
+            cluster_centers[1] = km.cluster_centers[1];
         }
-        cluster_centers[0] = km.cluster_centers[0];
-        cluster_centers[1] = km.cluster_centers[1];
         min_dist = km.min_dist;
         max_dist = km.max_dist;
         after = km.after;