cartodb/crankshaft
1
0
Fork 0
mirror of https://github.com/CartoDB/crankshaft.git synced 2024-11-01 10:20:48 +08:00
Code Issues Packages Projects Releases Wiki Activity

change parameter name to align with kmeans.spatial

Browse Source
This commit is contained in:
Andy Eschbacher 2017-01-10 09:52:46 -05:00
parent c6f64ad2f4
commit 69f38dd52e
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/py/crankshaft/crankshaft/clustering/kmeans.py
View File

@ -31,7 +31,7 @@ class Kmeans:
labels = km.fit_predict(zip(xs, ys))
return zip(ids, labels)
def nonspatial(self, subquery, colnames, num_clusters=5,
def nonspatial(self, subquery, colnames, no_clusters=5,
standardize=True, id_col='cartodb_id'):
"""
Inputs:
@ -43,7 +43,7 @@ class Kmeans:
'petal_width',
'sepal_length',
'petal_length']
num_clusters (int): number of clusters (greater than zero)
no_clusters (int): number of clusters (greater than zero)
id_col (string): name of the input id_column
Output:
@ -71,7 +71,7 @@ class Kmeans:
else:
cluster_columns = _extract_columns(data, len(colnames))
kmeans = KMeans(n_clusters=num_clusters,
kmeans = KMeans(n_clusters=no_clusters,
random_state=0).fit(cluster_columns)
centers = [json.dumps(dict(zip(colnames, c)))