import scala.math._object KMeansObject {
def main(args: Array[String]) : Unit = {
val dataset = Array (2,4,10,12,3,20,30,11,25)
val numberOfClusters = 2
var meanOfClusterOne = dataset(0)
var meanOfClusterTwo = dataset(8)
var clusterOne =new Array[Int](9)
var clusterTwo =new Array[Int](9)
var i = 0
var j = 0
var k = 0
for (i<-0 until 8 ){
if (abs(dataset(i) - meanOfClusterOne) < abs(dataset(i)-meanOfClusterTwo)){
clusterOne(j) += dataset(i)
j = j+1
}else{
clusterTwo(k) += dataset(i)
k = k+1
}
}
clusterOne(j) += meanOfClusterOne
clusterTwo(k) += meanOfClusterTwo
println(s" new mean of cluster one: $meanOfClusterOne")
println("dataset is:")
dataset.foreach(s => print(s" $s"))
println()
println("Cluster One is:")
clusterOne.foreach(s => print(s" $s"))
println()
println("Cluster Two is:")
clusterTwo.foreach(s => print(s" $s"))
println()
println(s"mean of cluster one: $meanOfClusterOne")
println(s"mean of cluster Two is: $meanOfClusterTwo")
println("clusterOne length is:" + clusterOne.length)
println("cluster Tow length is :" + clusterTwo.length)
println("dataset length is: " + dataset.length)
}
}
برچسب:
نویسنده: خنج