boyut uyumsuzluğu hatası kullanılarak Kıvılcım/MLLib kmeans aşağıdaki biçimde bir giriş dosyası var

Question

:

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

Ben dosyayı yüklenir ve Means eğitimli ve şimdi KMeansModel verilen tahmin işlevi kullanmak istiyorum ama Boyut uyuşmazlığı hatası alıyorum ve hata ayıklama hakkında nasıl gideceğimi bilmiyorum. Öngörme işlevinin girdisi hakkında herhangi bir belge görünmüyor.

kodu ...

Eğer model.predict komutuna "dizi" eklemeyi deneyebilir miyim

data = sc.textFile(join(movieLensHomeDir, "test.txt")) 
ratings = data.map(lambda line: [float(x.rstrip().lstrip().split('\r\n')[0]) for x in line.split(",")]) 
model = KMeans.train(ratings, 15, maxIterations=15, runs=10, initializationMode="random") 
labels = model.predict([0, 5, 3, 2, 4]) 

Answer 1

? , ör. Kmeans için

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

belgeler burada bulunabilir: http://spark.apache.org/docs/latest/api/python/pyspark.mllib.html#pyspark.mllib.clustering.KMeansModel