java ML回归预测_spark mllib 预测之LinearRegression(线性回归)

商品价格与消费者输入之间的关系

商品需求(y, 吨),价格(x1, 元),消费者收入(x2, 元)

y

x1

x2

5

1

1

8

1

2

7

2

1

13

2

3

18

3

4

建立需求函数: y = ax1+bx2

运行代码如下

package spark.regressionAnalysis

/**

*　线性回归, 建立商品价格与消费者输入之间的关系,

* 预测价格

*/

import org.apache.log4j.{Level, Logger}

import org.apache.spark.mllib.linalg.Vectors

import org.apache.spark.mllib.regression.{LabeledPoint, LinearRegressionWithSGD}

import org.apache.spark.{SparkConf, SparkContext}

object LinearRegression {

val conf = new SparkConf() //创建环境变量

.setMaster("local") //设置本地化处理

.setAppName("LinearRegression")//设定名称

val sc = new SparkContext(conf) //创建环境变量实例

def main(args: Array[