如何在sklearn库的k均值聚类中使用轮廓分数? [英] How to use silhouette score in k-means clustering from sklearn library?

问题描述

我想在脚本中使用轮廓分数，以自动计算来自sklearn的k均值聚类中的聚类数.

I'd like to use silhouette score in my script, to automatically compute number of clusters in k-means clustering from sklearn.

import numpy as np
import pandas as pd
import csv
from sklearn.cluster import KMeans
from sklearn.metrics import silhouette_score

filename = "CSV_BIG.csv"

# Read the CSV file with the Pandas lib.
path_dir = ".\\"
dataframe = pd.read_csv(path_dir + filename, encoding = "utf-8", sep = ';' ) # "ISO-8859-1")
df = dataframe.copy(deep=True)

#Use silhouette score
range_n_clusters = list (range(2,10))
print ("Number of clusters from 2 to 9: \n", range_n_clusters)

for n_clusters in range_n_clusters:
    clusterer = KMeans (n_clusters=n_clusters).fit(?)
    preds = clusterer.predict(?)
    centers = clusterer.cluster_centers_

    score = silhouette_score (?, preds, metric='euclidean')
    print ("For n_clusters = {}, silhouette score is {})".format(n_clusters, score)

有人可以帮助我解决问号吗?我不知道要问问而不是问号.我已经从一个示例中获取了代码. 带有注释的部分是以前的versione，在该版本中，我将k-means聚类进行了固定数目的聚类设置为4.这种方法是正确的，但是在我的项目中，我需要自动选择聚类的数目.

Someone can help me with question marks? I don't understand what to put instead of question marks. I have taken the code from an example. The commented part is the previous versione, where I do k-means clustering with a fixed number of clusters set to 4. The code in this way is correct, but in my project I need to automatically chose the number of clusters.

推荐答案

我假设您要进行轮廓比分以获得最佳编号.集群.

I am assuming you are going to silhouette score to get the optimal no. of clusters.

首先声明一个单独的对象KMeans，然后像这样在数据df上调用它的fit_predict函数

First declare a seperate object of KMeans and then call it's fit_predict functions over your data df like this

for n_clusters in range_n_clusters:
    clusterer = KMeans(n_clusters=n_clusters)
    preds = clusterer.fit_predict(df)
    centers = clusterer.cluster_centers_

    score = silhouette_score(df, preds)
    print("For n_clusters = {}, silhouette score is {})".format(n_clusters, score))

请参见 ="nofollow noreferrer">此官方示例.

这篇关于如何在sklearn库的k均值聚类中使用轮廓分数?的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！