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.
  1. Set up the package location where the program will reside:
package spark.ml.cookbook.chapter6
  1. Import the LogisticRegression package required to build and train the model. There are others forms of LogisticRegression in Spark MLlib, but for now we just concentrate on the basic logistic regression method:
import org.apache.spark.ml.classification.LogisticRegression
  1. Import SparkSession so we can gain access to the cluster, and Spark SQL, and hence the DataFrame and Dataset abstractions as needed via SparkSession:
org.apache.spark.sql.SparkSession
  1. Import the Vector Package from ml.linlang. This will allow ...

Get Apache Spark 2.x Machine Learning Cookbook now with O’Reilly online learning.

O’Reilly members experience live online training, plus books, videos, and digital content from 200+ publishers.