Where is the predict() in Logistic Regression of Spark MLLIb implemented?

Question

Can somebody point me to the implementation of predict() in LogisticRegressionModel of spark mllib? I could find a predictPoint() in the class LogisticRegressionModel, but where is predict()?

Answer 1

Since LogisticRegressionModel extends GeneralizedLinearModel with ClassificationModel (and some other less interesting stuff) it is reasonable to expect that predict is inherited from on of the above.

And as expected:

• predict(testData: RDD[Vector]): RDD[Double]
• def predict(testData: Vector): Double
• def predict(testData: JavaRDD[Vector]): JavaRDD[java.lang.Double]