python - 使用 Scikit Learn K-Means 大放异彩

Question

我正在尝试使 Blaze 数据对象适合 scikit kmeans 函数。

from blaze import *
from sklearn.cluster import KMeans
data_numeric = Data('data.csv')
data_cluster = KMeans(n_clusters=5)
data_cluster.fit(data_numeric)

数据样本:

A  B  C
1  32 34
5  57 92
89 67 21

它的抛出错误:

我已经能够使用 Pandas Dataframe 做到这一点。有什么方法可以将 blaze 对象提供给此函数？

Answer 1

我认为您需要在适合之前将 pandas 数据框转换为 numpy 数组。

from blaze import *
import numpy

from sklearn.cluster import KMeans
data_numeric = numpy.array(data('data.csv'))
data_cluster = KMeans(n_clusters=5)
data_cluster.fit(data_numeric)