MLlib中逻辑回归计算features的score代码

      val intercept = model.intercept
      val margin = dot(model.weights, features) + intercept
      val score = 1.0 / (1.0 + math.exp(-margin))

在源码里面默认的是如果score>0.5,则返回的label是1,否则是0

你可能感兴趣的:(机器学习)