Advertisement
Guest User

Untitled

a guest
Dec 5th, 2019
84
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.61 KB | None | 0 0
  1. def clusterUsingKmeans(data: DataFrame, k: Int, seed: Long):KMeansModel =
  2. {
  3. val kmeans = new KMeans().setK(k).setSeed(seed) //here setK refers to the centroids, and setSeed to whatever - its like that function in python.
  4. val model = kmeans.fit(data)
  5.  
  6. val pred = model.transform(data)
  7.  
  8. val evaluator = new ClusteringEvaluator()
  9.  
  10. val silhouette = evaluator.evaluate(pred)
  11. println(s"Silhouette with squared euclidean distance = $silhouette")
  12.  
  13. // Shows the result.
  14. println("Cluster Centers: ")
  15. model.clusterCenters.foreach(println)
  16.  
  17. model
  18. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement