代码之家 › 专栏 › 技术社区 › moirK

基于Scikit学习的混合数据集K均值聚类

one-hot-encoding categorical-data scikit-learn machine-learning python

-2

moirK · 技术社区 · 6 年前

我正在试验机器学习算法,有一个相当大的数据集,包含数字和分类数据。我在这里发布了以下帖子: http://www.ritchieng.com/machinelearning-one-hot-encoding/ 将分类特征编码为数字:

例如,我想尝试对整个数据集进行K-means聚类。我不知道如何使用我现在拥有的这个编码数据数组作为原始数据帧的一部分来运行机器学习算法。我真的很想举个例子。

1 回复 | 直到 6 年前

ekoulier 6 年前

我想你有一个热编码你的数据。为了使用K-means聚类,重新缩放数据很重要,因为您可能有一些数字特征将主导聚类。您可以尝试从 here (最著名的是MinMaxScaler和StandardScaler)。

之后,您可以参考 here 了解如何将KMeans与sklearn结合使用。一般而言,步骤如下:

导入KMeans:

from sklearn.cluster import KMeans

您实例化了一个KMeans对象,至少指定了集群的数量,这里我任意放置了8个:

kmeans = KMeans(n_clusters = 8)

然后用数据拟合对象(此处我的数据名为X):

kmeans.fit(X)

之后,您可以使用 .labels_ :

kmeans.labels_

您还可以使用 .predict :

kmeans.predict(new_X)

推荐文章

bz_jf · CNN训练损失太不稳定了

2 年前

ReactJs newbie · yolov4自定义培训,检测结果不正确

2 年前

Tushar Nautiyal · 我们需要在Flask应用程序中进行功能缩放吗

2 年前

Mahin Rahman · 我的培训和测试图表保持不变,有人能帮我解释一下,或者解释一下我哪里出错了?

2 年前

Mucida · BERT2:如何使用GPT2LMHeadModel开始一个句子,而不是完成它

2 年前

Bad Coder · 如何在Pyte中使用SMOTE?

2 年前

Sherwin R · 随机森林预测错误的输出形状

2 年前

Joseph · 重塑BatchDataset训练模型的输入-Tensorflow

2 年前

curiousninja · 如何从pandas中的特定列中删除非数值?

2 年前

Palkin Jangra · 如何迭代一列以获得每行的平均值?

2 年前