From fe2a10be75196559595ef0755e0cb07e8a635b1c Mon Sep 17 00:00:00 2001 From: BenFradet Date: Wed, 24 Jun 2015 20:31:49 +0200 Subject: [PATCH] callUDF => udf in ProbabilisticClassifier --- .../spark/ml/classification/ProbabilisticClassifier.scala | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/mllib/src/main/scala/org/apache/spark/ml/classification/ProbabilisticClassifier.scala b/mllib/src/main/scala/org/apache/spark/ml/classification/ProbabilisticClassifier.scala index 330ae2938f4e0..15e23466250b3 100644 --- a/mllib/src/main/scala/org/apache/spark/ml/classification/ProbabilisticClassifier.scala +++ b/mllib/src/main/scala/org/apache/spark/ml/classification/ProbabilisticClassifier.scala @@ -113,9 +113,9 @@ private[spark] abstract class ProbabilisticClassificationModel[ } if ($(predictionCol).nonEmpty) { val predUDF = if ($(rawPredictionCol).nonEmpty) { - callUDF(raw2prediction _, DoubleType, col($(rawPredictionCol))) + udf[Double, Vector](raw2prediction).apply(col($(rawPredictionCol))) } else if ($(probabilityCol).nonEmpty) { - callUDF(probability2prediction _, DoubleType, col($(probabilityCol))) + udf[Double, Vector](probability2prediction).apply(col($(probabilityCol))) } else { callUDF(predict _, DoubleType, col($(featuresCol))) }