callUDF => udf in Classifier where possible

BenFradet · BenFradet · commit 0ea30b36702d · 2015-06-24T20:07:13.000+02:00
diff --git a/mllib/src/main/scala/org/apache/spark/ml/classification/Classifier.scala b/mllib/src/main/scala/org/apache/spark/ml/classification/Classifier.scala
@@ -108,7 +108,7 @@ abstract class ClassificationModel[FeaturesType, M <: ClassificationModel[Featur
     }
     if (getPredictionCol != "") {
       val predUDF = if (getRawPredictionCol != "") {
-        callUDF(raw2prediction _, DoubleType, col(getRawPredictionCol))
+        udf[Double, Vector](raw2prediction).apply(col(getRawPredictionCol))
       } else {
         callUDF(predict _, DoubleType, col(getFeaturesCol))
       }

Original file line number	Diff line number	Diff line change
`@@ -108,7 +108,7 @@ abstract class ClassificationModel[FeaturesType, M <: ClassificationModel[Featur`
`108`	`108`	`}`
`109`	`109`	`if (getPredictionCol != "") {`
`110`	`110`	`val predUDF = if (getRawPredictionCol != "") {`
`111`		`- callUDF(raw2prediction _, DoubleType, col(getRawPredictionCol))`
	`111`	`+ udf[Double, Vector](raw2prediction).apply(col(getRawPredictionCol))`
`112`	`112`	`} else {`
`113`	`113`	`callUDF(predict _, DoubleType, col(getFeaturesCol))`
`114`	`114`	`}`