working algorithm

src/main/scala/KMeans.scala

@@ -45,7 +45,7 @@ object KMeans {
     val newCentres = clusterMap.reduceByKey((a,b) => getMeanVector(a,b,m))
     val arrayNewCentres = newCentres.collect()
 
-    var results = new Array[Array[Float]](m)
+    var results = new Array[Array[Float]](K)
     for ((i,x) <- arrayNewCentres) {
       results(i) = x
     }

src/main/scala/Main.scala

@@ -30,7 +30,7 @@ object Main {
     // get the users XML file
 
     val users = df("users")
-    val centres = KMeans.train(users, 6)
+    val centres = KMeans.train(users, 2)
     //val centresArray = centres.collect()
     //val unwrap = centresArray.map(x => x._2)
     //unwrap.foreach(println)