当前位置:   article > 正文

逻辑回归简单的代码(scala实现)_scala逻辑回归案例

scala逻辑回归案例
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}))
声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/木道寻08/article/detail/737137
推荐阅读
  

闽ICP备14008679号