900字范文,内容丰富有趣,生活中的好帮手!
900字范文 > spark mllib 预测之LinearRegression(线性回归)

spark mllib 预测之LinearRegression(线性回归)

时间:2024-03-18 05:58:10

相关推荐

spark mllib 预测之LinearRegression(线性回归)

为什么80%的码农都做不了架构师?>>>

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

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

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

运行代码如下

package spark.regressionAnalysis/***线性回归, 建立商品价格与消费者输入之间的关系,* 预测价格*/import org.apache.log4j.{Level, Logger}import org.apache.spark.mllib.linalg.Vectorsimport 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[String]) {val data = sc.textFile("./src/main/spark/regressionAnalysis/lr.txt")//获取数据集路径val parsedData = data.map { line => //开始对数据集处理val parts = line.split('|') //根据逗号进行分区LabeledPoint(parts(0).toDouble, Vectors.dense(parts(1).split(',').map(_.toDouble)))}.cache() //转化数据格式//LabeledPoint,numIterations, stepSizeval model = LinearRegressionWithSGD.train(parsedData, 2, 0.1) //建立模型val result = model.predict(Vectors.dense(1, 3))//通过模型预测模型println(model.weights)println(model.weights.size)println(result)//打印预测结果}}

lr.txt

5|1,18|1,27|2,113|2,318|3,4

结果如图

本内容不代表本网观点和政治立场,如有侵犯你的权益请联系我们处理。
网友评论
网友评论仅供其表达个人看法,并不表明网站立场。