callUDF => udf in ProbabilisticClassifier

BenFradet · BenFradet · commit fe2a10be7519 · 2015-06-24T20:31:53.000+02:00
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
@@ -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)))
       }

Original file line number	Diff line number	Diff line change
`@@ -113,9 +113,9 @@ private[spark] abstract class ProbabilisticClassificationModel[`
`113`	`113`	`}`
`114`	`114`	`if ($(predictionCol).nonEmpty) {`
`115`	`115`	`val predUDF = if ($(rawPredictionCol).nonEmpty) {`
`116`		`- callUDF(raw2prediction _, DoubleType, col($(rawPredictionCol)))`
	`116`	`+ udf[Double, Vector](raw2prediction).apply(col($(rawPredictionCol)))`
`117`	`117`	`} else if ($(probabilityCol).nonEmpty) {`
`118`		`- callUDF(probability2prediction _, DoubleType, col($(probabilityCol)))`
	`118`	`+ udf[Double, Vector](probability2prediction).apply(col($(probabilityCol)))`
`119`	`119`	`} else {`
`120`	`120`	`callUDF(predict _, DoubleType, col($(featuresCol)))`
`121`	`121`	`}`