1. 程式人生 > >MLlib中邏輯迴歸計算features的score程式碼

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