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

xiaoxiao2021-02-28  108

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

转载请注明原文地址: https://www.6miu.com/read-61765.html

最新回复(0)