O'Reilly logo

Fast Data Processing with Spark 2 - Third Edition by Krishna Sankar

Stay ahead with the world's most comprehensive technology and business learning platform.

With Safari, you learn the way you learn best. Get unlimited access to videos, live online training, learning paths, books, tutorials, and more.

Start Free Trial

No credit card required

Clustering

Spark MLlib has implemented the K-means clustering algorithm. The model training and prediction interfaces are similar to other machine learning algorithms. Let's see how it works by going through an example.

Let's use a sample data that has two dimensions: x and y. The plot of the points looks like the following screenshot:

Clustering

From the preceding graph, we can see that four clusters form one solution. Let's try k = 2 and k=4. Let's see how the Spark clustering algorithm handles this Dataset and the groupings.

Loading data

By now, we know very well how to load data using the read.csv() method. The following code file is ML03v2.scala:

We run ...

With Safari, you learn the way you learn best. Get unlimited access to videos, live online training, learning paths, books, interactive tutorials, and more.

Start Free Trial

No credit card required