diff --git a/src/py/crankshaft/crankshaft/clustering/kmeans.py b/src/py/crankshaft/crankshaft/clustering/kmeans.py index d070bf9..383584e 100644 --- a/src/py/crankshaft/crankshaft/clustering/kmeans.py +++ b/src/py/crankshaft/crankshaft/clustering/kmeans.py @@ -65,7 +65,7 @@ def kmeans_nonspatial(query, colnames, num_clusters=5, cluster_columns = _scale_data( _extract_columns(db_resp, out_id_colname)) else: - cluster_columns = _extract_columns(db_resp, id_col=out_id_colname) + cluster_columns = _extract_columns(db_resp, out_id_colname) # TODO: decide on optimal parameters for most cases # Are there ways of deciding parameters based on inputs?