/Linear-Polynomial-Regression-Scala-SparkML

This is an example of Linear Regression done in SparkML and using the class PolynomialExpansion.

Primary LanguageScala

Linear-Polynomial-Regression-Scala-SparkML

PolyLinearRegression.scala : This is an example of Linear Regression done in SparkML and using the class PolynomialExpansion.

NormalEquation.scala : Example of normal equation in SparkML with Scala, using DenseMatrix, DenseVector and RowMatrix

Tools.scala : Various functions used in the classes.

Among them :

Function which converts DataFrame to DenseMatrix

Function which converts DataFrame to DenseVector

Function which converts DataFrame to RDD etc. ...