O'Reilly logo

Apache Spark for Data Science Cookbook by Padma Priya Chitturi

Stay ahead with the world's most comprehensive technology and business learning platform.

With Safari, you learn the way you learn best. Get unlimited access to videos, live online training, learning paths, books, tutorials, and more.

Start Free Trial

No credit card required

Applying linear regression

In this recipe, we'll see how to apply linear regression.

Getting ready

To step through this recipe, you will need a running Spark cluster in any one of the modes, for instance, local, standalone, YARN, or Mesos. For installing Spark on a standalone cluster, please refer to http://spark.apache.org/docs/latest/spark-standalone.html. Also, include the Spark MLlib package in the build.sbt file so that it downloads the related libraries and the API can be used. Install Hadoop (optionally), Scala, and Java.

How to do it…

  1. The final step is to convert data into train and test datasets and apply the regression model on the training data:
     //Split the data frame into train and test sets val train_Df = final_Df.filter(final_Df("Item_Outlet_Sales").isNotNull) ...

With Safari, you learn the way you learn best. Get unlimited access to videos, live online training, learning paths, books, interactive tutorials, and more.

Start Free Trial

No credit card required