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({