使用Spark/MLLib Kmeans的维度不匹配错误

Question

我有以下格式的输入文件：

1, 4, 5, 0, 3
0, 4, 5, 1, 4
.....

我加载了文件并训练了Means，现在想使用KMeansModel中给出的predict函数，但是我得到了一个维度不匹配的错误，我不确定如何进行调试。似乎没有任何关于predict函数输入的文档。在

代码。。。在

^{pr2}$

Answer 1

你能试着给你的模型.预测指挥部？ e、 g

labels = model.predict((array([0, 5, 3, 2, 4]))

KMeans文档可在以下位置找到： http://spark.apache.org/docs/latest/api/python/pyspark.mllib.html#pyspark.mllib.clustering.KMeansModel