逻辑回归简单的代码（scala实现）

最新推荐文章于 2022-12-25 01:16:06 发布

wyn1175397098zt

最新推荐文章于 2022-12-25 01:16:06 发布

阅读量1.1k

点赞数

分类专栏：机器学习算法

本文链接：https://blog.csdn.net/wyn1175397098zt/article/details/79813841

版权

这篇博客介绍了如何使用Scala编程语言实现逻辑回归算法。通过lpsa.data数据集，展示了数据的格式，其中包含目标变量y和一系列特征x的值。

摘要由CSDN通过智能技术生成

package com.sxt.scala.lr

import org.apache.spark.SparkConf
import org.apache.spark.SparkContext
import org.apache.spark.mllib.regression.LabeledPoint
import org.apache.spark.mllib.linalg.Vectors
import org.apache.spark.mllib.regression.LinearRegressionWithSGD
import org.apache.spark.mllib.regression.LinearRegressionModel
import org.apache.spark.sql.SQLContext

object MyLinearRegression {
def main(args:Array[String]){
    val conf = new SparkConf().setAppName("mylinear").setMaster("local")
    val sc = new SparkContext(conf)

    val examples = sc.textFile("lpsa.data").map {
      line =>
        val parts = line.split(",")
        val y = parts(0)
        val xs = parts(1)
        LabeledPoint(parts(0).toDouble,Vectors.dense(parts(1).split(" ").map { _.toDouble}))