From 8b8c216f3d939bb8289ae7442a7282c0f46ffc51 Mon Sep 17 00:00:00 2001 From: quillotm Date: Tue, 24 Aug 2021 19:52:12 +0200 Subject: [PATCH] Fixed issues, k was equal to 5 instead of the k parameter --- volia/clustering_modules/kmeans_multidistance.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/volia/clustering_modules/kmeans_multidistance.py b/volia/clustering_modules/kmeans_multidistance.py index 2d2971d..5ce8e42 100644 --- a/volia/clustering_modules/kmeans_multidistance.py +++ b/volia/clustering_modules/kmeans_multidistance.py @@ -66,7 +66,7 @@ class kmeansMultidistance(): # Learning k-means model results = [] for i in range(ninit): - model = KMeans(k=5, + model = KMeans(k=k, maxiter=maxiter, distance=self.distance, record_heterogeneity=[], -- 1.8.2.3