How to do it...

  1. Start a new project in IntelliJ or in an IDE of your choice. Make sure the necessary JAR files are included.
  1. We define the package information for the Scala program:
package spark.ml.cookbook.chapter7
  1. Import the necessary packages:
import java.text.DecimalFormat import org.apache.log4j.{Level, Logger} import org.apache.spark.sql.SparkSession import org.jfree.chart.{ChartFactory, ChartFrame, JFreeChart} import org.jfree.chart.axis.NumberAxis import org.jfree.chart.plot.PlotOrientation import org.jfree.data.xy.{XYSeries, XYSeriesCollection}
  1. We now define a Scala case class to model the ratings data:
case class Rating(userId: Int, movieId: Int, rating: Float, timestamp: Long)
  1. Let's define a function to display a JFreeChart ...

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.