在Sparks Mllib中使用哪个函数进行预测

Question

使用什么函数来计算某个数据点的预测。请考虑以下代码：

final LogisticRegressionModel model = logisticRegression.run(train.rdd());
point = train.rdd().take(0);
Double prediction = model.predict(point.features());

我以为会是

但我找不到实现此功能的文档。

Answer 1

它使用标准逻辑函数（公式中的 y 不合适）：<​​/ p>

is implemented as：

val margin = dot(weightMatrix, dataMatrix) + intercept
val score = 1.0 / (1.0 + math.exp(-margin))

要获得原始价值，您应该clearThreshold。