def cluster(X, eps=1, min_pts=30, algorithm='DBSCAN', n_clusters=10):
if algorithm == 'DBSCAN':
cluster_result = DBSCAN(eps=eps, min_samples=min_pts).fit(X)
elif algorithm == 'KMeans':
cluster_result = KMeans(n_clusters=n_clusters)
labels = cluster_result.labels_
return labels
评论列表
文章目录