Apache Spark 2.x Machine Learning Cookbook

How to do it...

1. Start a new project in IntelliJ or in an IDE of your choice. Make sure that the necessary JAR files are included.
2. Set up the package location where the program will reside:

package spark.ml.cookbook.chapter4

3. Import the necessary packages for SparkContext to get access to the cluster:

import org.apache.spark.mllib.linalg.Vectorsimport org.apache.spark.sql.SparkSessionimport org.apache.spark.mllib.clustering.KMeans

4. Create Spark's configuration and SparkContext:

val spark = SparkSession.builder.master("local[*]")   // if use cluster master("spark://master:7077").appName("myPMMLExport").config("spark.sql.warehouse.dir", ".").getOrCreate()

5. We read the data from a text file; the data file contains a sample dataset for a KMeans ...