diff --git a/volia/clustering.py b/volia/clustering.py index 1dfae8a..9d8e5c9 100644 --- a/volia/clustering.py +++ b/volia/clustering.py @@ -138,7 +138,7 @@ def kmeans_run(features: str, mkdir(output) for k in klist: k = int(k) - fit_model(k, path.join(output, "clustering_" + str(i) + ".pkl")) + fit_model(k, path.join(output, "clustering_" + str(k) + ".pkl")) print(json.dumps(json_content))