chavalitros
/
IN-P2
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Save the number of clusters correctly

This commit is contained in:
coolneng 2020-12-11 17:17:46 +01:00
parent 6fe18d594c
commit 7e356fed37
Signed by: coolneng
GPG Key ID: 9893DA236405AF57
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/P2/processing.py
View File

@ -41,7 +41,7 @@ def predict_data(data, model, results, sample):
df=results, df=results,
model=model_name, model=model_name,
prediction=prediction, prediction=prediction,
clusters=len(prediction), clusters=len(set(prediction)),
calinski=calinski, calinski=calinski,
silhouette=silhouette, silhouette=silhouette,
time=execution_time, time=execution_time,