print output correctly
This commit is contained in:
@@ -35,7 +35,7 @@ object KMeans {
|
||||
for (j <- 0 until m) {
|
||||
centres(i)(j) += i+j
|
||||
}
|
||||
println("centre " + i + " = " + centres(i) )
|
||||
println("centre " + i + " = " + centres(i).mkString("[",",","]") )
|
||||
}
|
||||
|
||||
/*var centres : ArrayBuffer[Float] = new ArrayBuffer[Float](K)
|
||||
@@ -53,8 +53,9 @@ object KMeans {
|
||||
|
||||
for (i <- 0 until iterations) {
|
||||
centres = clustering(centres, rowsAsArray, m, K)
|
||||
for (i <- 0 until K) {
|
||||
println("centre " + i + " = " + centres(i) )
|
||||
println("\niteration " + i + " :")
|
||||
for (j <- 0 until K) {
|
||||
println("centre " + j + " = " + centres(j).mkString("[",",","]") )
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@@ -16,6 +16,7 @@ object Main {
|
||||
// spark cluster
|
||||
val sc = new SparkContext(new SparkConf().setAppName("Spark KMeans Clustering"))
|
||||
val sqlContext = new org.apache.spark.sql.SQLContext(sc)
|
||||
sc.setLogLevel("WARN") //reduce spark command line verbosity
|
||||
|
||||
import sqlContext.implicits._
|
||||
|
||||
@@ -29,7 +30,7 @@ object Main {
|
||||
// get the users XML file
|
||||
|
||||
val users = df("users")
|
||||
val centres = KMeans.train(users, 3)
|
||||
val centres = KMeans.train(users, 6)
|
||||
//val centresArray = centres.collect()
|
||||
//val unwrap = centresArray.map(x => x._2)
|
||||
//unwrap.foreach(println)
|
||||
|
||||
Reference in New Issue
Block a user