diff --git a/ml/kmeans.v b/ml/kmeans.v index cc30f1968..9e869916a 100644 --- a/ml/kmeans.v +++ b/ml/kmeans.v @@ -120,6 +120,7 @@ pub fn (mut o Kmeans) compute_centroids() { } pub struct TrainConfig { +pub: epochs int tol_norm_change f64 } diff --git a/ml/knn.v b/ml/knn.v index 6d95a0375..407cbfec0 100644 --- a/ml/knn.v +++ b/ml/knn.v @@ -109,9 +109,9 @@ pub fn (mut knn KNN) train() { // data needed for KNN.predict pub struct PredictConfig { +pub: max_iter int k int -mut: to_pred []f64 }