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 the O’Reilly learning platform.

O’Reilly members experience books, live events, courses curated by job role, and more from O’Reilly and nearly 200 top publishers.