基于Spark/Scala的Logistic Regression算法_scala不调用库函数实现逻辑回归

作者：神奇cpp | 2024-06-19 16:59:16

踩

scala不调用库函数实现逻辑回归

一个基于Spark/Scala的迭代式逻辑回归（Logistic Regression）算法-----寻找最优分类面来将多维特征空间中的两类点集分开：

val points = spark.textFile(...).map(parsePoint).cache()
var w = Vector.random(D) // current separating plane
for (i <- 1 to ITERATIONS) {
  val gradient = points.map(p =>
    (1 / (1 + exp(-p.y*(w dot p.x))) - 1) * p.y * p.x
  ).reduce(_ + _)
  w -= gradient
}
println("Final separating plane: " + w)

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/神奇cpp/article/detail/737110