0

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]])

2 つの新しいクラスターを変数としてデータに追加したいと考えています。私が使用したコードは次のとおりです。

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()

それらを変数として保存するにはどうすればよいですか? シルエット係数など、これらのクラスターの品質を評価する機能はありますか?

4

1 に答える 1