adds silhouettes to output

2016-11-14 23:29:38 +00:00
parent b6dae5e380
commit af536757fe
2 changed files with 3 additions and 2 deletions
@@ -76,11 +76,12 @@ def kmeans_nonspatial(query, colnames, num_clusters=5,
               for c in kmeans.cluster_centers_[kmeans.labels_]]

    silhouettes = metrics.silhouette_samples(cluster_columns,
-                                             labels,
+                                             kmeans.labels_,
                                             metric='sqeuclidean')

    return zip(kmeans.labels_,
               centers,
+               silhouettes,
               db_resp[0][out_id_colname])