小编use*_*415的帖子

Spark MlLib线性回归(线性最小二乘)给出随机结果

我是火花和机器学习的新手.我已经成功完成了一些Mllib教程,我无法使用这个教程:

我在这里找到了示例代码:https: //spark.apache.org/docs/latest/mllib-linear-methods.html#linear-least-squares-lasso-and-ridge-regression

(部分LinearRegressionWithSGD)

这是代码:

import org.apache.spark.mllib.regression.LabeledPoint
import org.apache.spark.mllib.regression.LinearRegressionModel
import org.apache.spark.mllib.regression.LinearRegressionWithSGD
import org.apache.spark.mllib.linalg.Vectors

// Load and parse the data
val data = sc.textFile("data/mllib/ridge-data/lpsa.data")
val parsedData = data.map { line =>
  val parts = line.split(',')
  LabeledPoint(parts(0).toDouble, Vectors.dense(parts(1).split(' ').map(_.toDouble)))
}.cache()

// Building the model
val numIterations = 100
val model = LinearRegressionWithSGD.train(parsedData, numIterations)

// Evaluate model on training examples and compute training error
val valuesAndPreds = parsedData.map { point =>
  val prediction = model.predict(point.features)
  (point.label, prediction)
}
val MSE = …
Run Code Online (Sandbox Code Playgroud)

machine-learning apache-spark apache-spark-mllib

5
推荐指数
1
解决办法
6582
查看次数