Save the number of clusters correctly
This commit is contained in:
parent
6fe18d594c
commit
7e356fed37
|
@ -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,
|
||||
|
|
Loading…
Reference in New Issue