From 2edadabee352868100d1829adb0c1dfb9eb6b4ad Mon Sep 17 00:00:00 2001 From: quillotm Date: Mon, 23 Aug 2021 21:21:14 +0200 Subject: [PATCH] By default, kmeans mahalanobis is with constrains --- volia/clustering.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/volia/clustering.py b/volia/clustering.py index 9d8e5c9..af8aefc 100644 --- a/volia/clustering.py +++ b/volia/clustering.py @@ -98,7 +98,7 @@ def kmeans_run(features: str, if mahalanobis: if debug: print("Mahalanobis activated") - model = CLUSTERING_METHODS["k-means-mahalanobis"] + model = CLUSTERING_METHODS["k-means-mahalanobis-constrained"] model.fit(X, k, tol, ninit, maxiter, debug) model.save(output_file) json_content["models"].append({ -- 1.8.2.3