当前位置:   article > 正文

spark实现下的逻辑回归(logistic regression)_spark logisticregression

spark logisticregression

在spark2.0之前,逻辑回归只能做二分类的。之后加上了多分类。

根据我的理解,逻辑回归既可以用来分类,也可以用来回归,但是官网并没有给出logistic regression回归的例子,只有线性回归(linear regression)的例子

此外,官网给出了逻辑回归二分类和多分类的例子,但是多分类中的setFamlily方法无法在spark1.6和spark2.0的版本上运行,无论是用IDEA还是spark-shell。

这里给出完整的逻辑回归二分类代码,数据创建方式有两种:手动代码输入和从文件读取。

代码如下:

  1. /**
  2. * Created by wangtuntun on 17-3-7.
  3. * 将数据代入模型进行预测
  4. * 本来想用logistic regression做回归的,结果点开的是官网的classification下的代码
  5. * 而且还是只支持二分类的代码
  6. */
  7. import org.apache.spark.{SparkContext, SparkConf}
  8. import org.apache.spark.ml.classification.LogisticRegression
  9. import org.apache.spark.ml.param.ParamMap
  10. import org.apache.spark.mllib.linalg.{Vector, Vectors}
  11. import org.apache.spark.sql.Row
  12. import org.apache.spark.sql.SQLContext
  13. object predict_with_logistic_regression_classification {
  14. def main(args: Array[String]): Unit = {
  15. // Prepare training data from a
声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/笔触狂放9/article/detail/737098
推荐阅读
相关标签
  

闽ICP备14008679号