Commit 2edadabee352868100d1829adb0c1dfb9eb6b4ad
1 parent
55bcf758f3
Exists in
master
By default, kmeans mahalanobis is with constrains
Showing 1 changed file with 1 additions and 1 deletions Side-by-side Diff
volia/clustering.py
... | ... | @@ -98,7 +98,7 @@ |
98 | 98 | if mahalanobis: |
99 | 99 | if debug: |
100 | 100 | print("Mahalanobis activated") |
101 | - model = CLUSTERING_METHODS["k-means-mahalanobis"] | |
101 | + model = CLUSTERING_METHODS["k-means-mahalanobis-constrained"] | |
102 | 102 | model.fit(X, k, tol, ninit, maxiter, debug) |
103 | 103 | model.save(output_file) |
104 | 104 | json_content["models"].append({ |