Apache Spark 2.x Machine Learning Cookbook

There's more...

To ensure completeness, we include the import statements 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};