Plot Kmeans TruncatedSVD Python之后的集群

Question

我试图在我的数据集上绘制运行群集的结果，但是我得到了错误：

  File "cluster.py", line 93, in <module>
    Z = kmeans.predict(np.c_[xx.ravel(), yy.ravel()])
  File "/usr/local/lib/python2.7/dist-packages/sklearn/cluster/k_means_.py", line 957, in predict
    X = self._check_test_data(X)
  File "/usr/local/lib/python2.7/dist-packages/sklearn/cluster/k_means_.py", line 867, in _check_test_data
    n_features, expected_n_features))
ValueError: Incorrect number of features. Got 2 features, expected 73122

我对fit()的调用工作正常，但绘图是出问题的地方。在

我的代码是：

^{pr2}$

有人能建议我如何修改代码以获得集群的图表吗？在

Answer 1

回溯告诉你问题出在哪里：

ValueError: Incorrect number of features. Got 2 features, expected 73122

kmeans分类器适合73122维训练样本，因此不能使用kmeans对2维测试样本进行预测。在

要修复代码，只需将kmeans.fit(X)更改为kmeans.fit(reduced_data)。在