From 7e356fed37b4f5b342218b035ad0fbaef39582f5 Mon Sep 17 00:00:00 2001
From: coolneng <akasroua@gmail.com>
Date: Fri, 11 Dec 2020 17:17:46 +0100
Subject: [PATCH] Save the number of clusters correctly

---
 src/P2/processing.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/P2/processing.py b/src/P2/processing.py
index 44fffaf..cb64fa2 100644
--- 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,