Github user mgaido91 commented on a diff in the pull request:
https://github.com/apache/spark/pull/19676#discussion_r155929522
--- Diff:
examples/src/main/java/org/apache/spark/examples/ml/JavaKMeansExample.java ---
@@ -51,9 +52,14 @@ public static void main(String[] args) {
KMeans kmeans = new KMeans().setK(2).setSeed(1L);
KMeansModel model = kmeans.fit(dataset);
- // Evaluate clustering by computing Within Set Sum of Squared Errors.
- double WSSSE = model.computeCost(dataset);
- System.out.println("Within Set Sum of Squared Errors = " + WSSSE);
+ // Make predictions
+ Dataset<Row> predictions = model.transform(dataset);
+
+ // Evaluate clustering by computing Silhouette score
+ ClusteringEvaluator evaluator = new ClusteringEvaluator();
+
+ double silhouette = evaluator.evaluate(predictions);
+ System.out.println("Silhouette with squared euclidean distance = " +
silhouette);
--- End diff --
Thanks, I don't think I am changing the code again, but I can fix this
grammatical error if you want.
---
---------------------------------------------------------------------
To unsubscribe, e-mail: reviews-unsubscr...@spark.apache.org
For additional commands, e-mail: reviews-h...@spark.apache.org
