Updated constructors

vlang · Oct 17, 2023 · a3be2ce · a3be2ce
1 parent 740dfed
commit a3be2ce
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/ml/kmeans.v b/ml/kmeans.v
@@ -14,6 +14,7 @@ mut:
 	stat       &Stat[f64] // statistics about x (data)
 	nb_classes int        // expected number of classes
 	bins       &gm.Bins   // "bins" to speed up searching for data points given their coordinates (2D or 3D only at the moment)
+	nb_iter    int        // number of iterations
 pub mut:
 	classes    []int   // [nb_samples] indices of classes of each sample
 	centroids  [][]f64 // [nb_classes][nb_features] coordinates of centroids
@@ -131,6 +132,7 @@ pub fn (mut o Kmeans) train(config TrainConfig) {
 		o.compute_centroids()
 		nb_iter++
 	}
+	o.nb_iter = o.nb_iter + nb_iter
 }
 
 // str is a custom str function for observers to avoid printing data