Save the number of clusters correctly

2020-12-11 17:17:46 +01:00
parent 6fe18d594c
commit 7e356fed37
1 changed files with 1 additions and 1 deletions
--- a/src/P2/processing.py
+++ b/src/P2/processing.py
@@ -41,7 +41,7 @@ def predict_data(data, model, results, sample):
        df=results,
        model=model_name,
        prediction=prediction,
-        clusters=len(prediction),
+        clusters=len(set(prediction)),
        calinski=calinski,
        silhouette=silhouette,
        time=execution_time,