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

来源:互联网 发布:淘宝买家评价多久清空 编辑:程序博客网 时间:2024/06/05 15:01
      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

原创粉丝点击