如:
kmeans = KMeans(n_clusters=2, random_state=0).fit(tr_sfa[:,1])
这是因为在最新版本的sklearn中,所有的数据都应该是二维矩阵,哪怕它只是单独一行或一列。
解决添加.reshape(-1,1)即可
kmeans = KMeans(n_clusters=2, random_state=0).fit(tr_sfa[:,1].reshape(-1,1))
如:
kmeans = KMeans(n_clusters=2, random_state=0).fit(tr_sfa[:,1])
这是因为在最新版本的sklearn中,所有的数据都应该是二维矩阵,哪怕它只是单独一行或一列。
解决添加.reshape(-1,1)即可
kmeans = KMeans(n_clusters=2, random_state=0).fit(tr_sfa[:,1].reshape(-1,1))