如何在sklearn库的k-means聚类中使用轮廓分数?

我想在我的脚本中使用轮廓分数来自动计算sklearn库中k-means聚类的簇数。

有人能帮我解决这些问号吗?我不明白应该用什么来替换这些问号。我是从一个例子中拿到的代码。注释部分是之前的版本,其中我使用固定簇数4进行k-means聚类。这样代码是正确的,但在我的项目中,我需要自动选择簇数。


回答:

我假设您将使用轮廓分数来获取最佳簇数。

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

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("当簇数为{}时,轮廓分数为{})".format(n_clusters, score))

请查看这个官方示例以获得更多清晰的说明。

Related Posts

使用LSTM在Python中预测未来值

这段代码可以预测指定股票的当前日期之前的值,但不能预测…

如何在gensim的word2vec模型中查找双词组的相似性

我有一个word2vec模型,假设我使用的是googl…

dask_xgboost.predict 可以工作但无法显示 – 数据必须是一维的

我试图使用 XGBoost 创建模型。 看起来我成功地…

ML Tuning – Cross Validation in Spark

我在https://spark.apache.org/…

如何在React JS中使用fetch从REST API获取预测

我正在开发一个应用程序,其中Flask REST AP…

如何分析ML.NET中多类分类预测得分数组?

我在ML.NET中创建了一个多类分类项目。该项目可以对…

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注