我通过 dbscan skelearn 制作了以下集群
我的数据是一个 numpy 数组:
array([[-0.22725194, -0.68548221],
[ 0.01525107, -0.98825191],
[-0.29117618, -0.69614647],
...,
[ 0.62125361, -0.79422623],
[ 0.59627969, -0.82673572],
[ 0.58919524, -1.04003462]])
我想将两个新集群作为变量添加到我的数据中。我使用的代码是:
from sklearn.cluster import DBSCAN
data3 = np.array(data3)
dbscan = DBSCAN(random_state=111, eps=0.3)
dbscan.fit(data3)
# visualization
from sklearn.decomposition import PCA
pca = PCA(n_components=2).fit(data3)
pca_2d = pca.transform(data3)
for i in range(0, pca_2d.shape[0]):
if dbscan.labels_[i] == 0:
c1 = plt.scatter(pca_2d[i,0],pca_2d[i,1],c='r',
marker='+')
elif dbscan.labels_[i] == 1:
c2 = plt.scatter(pca_2d[i,0],pca_2d[i,1],c='g',
marker='o')
elif dbscan.labels_[i] == -1:
c3 = plt.scatter(pca_2d[i,0],pca_2d[i,1],c='b',
marker='*')
plt.legend([c1, c2, c3], ['Cluster 1', 'Cluster 2',
'Noise'])
plt.title('DBSCAN finds 2 clusters and noise')
plt.show()
如何将它们保存为变量?是否有评估这些集群质量的函数,例如 Silhoutte Coefficient?