本文整理汇总了Scala中org.apache.spark.ml.classification.LogisticRegressionModel类的典型用法代码示例。如果您正苦于以下问题：Scala LogisticRegressionModel类的具体用法？Scala LogisticRegressionModel怎么用？Scala LogisticRegressionModel使用的例子？那么恭喜您, 这里精选的类代码示例或许可以为您提供帮助。
在下文中一共展示了LogisticRegressionModel类的1个代码示例，这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞，您的评价将有助于系统推荐出更棒的Scala代码示例。
示例1: LogisticRegressionHelpers
//设置package包名称以及导入依赖的类
package com.zobot.ai.spark.helpers
import org.apache.spark.ml.classification.{LogisticRegression, LogisticRegressionModel}
import org.apache.spark.ml.linalg.Vector
import org.apache.spark.sql.{DataFrame, Row}
object LogisticRegressionHelpers {
case class ModelTestResults (
features: Vector,
label: Double,
probability: Vector,
prediction: Double
)
def trainModel(estimator: LogisticRegression, trainingSet: DataFrame): LogisticRegressionModel = estimator.fit(trainingSet)
def testModel(transformer: LogisticRegressionModel, testSet: DataFrame): LogisticRegressionModel = {
transformer.transform(testSet).select("features", "label", "myProbability", "prediction").collect().foreach {
case Row(features: Vector, label: Double, prob: Vector, prediction: Double) =>
ModelTestResults(features, label, prob, prediction)
}
transformer
}
}