Apache Spark 2.x Machine Learning Cookbook

There's more...

To ensure completeness, we include the import statement that we used prior to Spark 2.0.0 to run the code (namely, Spark 1.5.2):

import org.apache.spark._
import org.apache.spark.rdd.RDD 
import org.apache.spark.sql.SQLContext 
import org.apache.spark.mllib.linalg 
import org.apache.spark.util 
import Array._
import org.apache.spark.sql._
import org.apache.spark.sql.types 
import org.apache.spark.sql.DataFrame 
import org.apache.spark.sql.Row; 
import org.apache.spark.sql.types.{ StructType, StructField, StringType};