fix transform return DF

yanboliang · yanboliang · commit e5aea09d3eed · 2016-04-12T07:02:22.000-07:00
diff --git a/mllib/src/main/scala/org/apache/spark/ml/regression/GeneralizedLinearRegression.scala b/mllib/src/main/scala/org/apache/spark/ml/regression/GeneralizedLinearRegression.scala
@@ -730,7 +730,7 @@ class GeneralizedLinearRegressionModel private[ml] (
     if ($(linkPredictionCol).nonEmpty) {
       output = output.withColumn($(linkPredictionCol), predictLinkUDF(col($(featuresCol))))
     }
-    output
+    output.toDF
   }
 
   private var trainingSummary: Option[GeneralizedLinearRegressionSummary] = None

Original file line number	Diff line number	Diff line change
`@@ -730,7 +730,7 @@ class GeneralizedLinearRegressionModel private[ml] (`
`730`	`730`	`if ($(linkPredictionCol).nonEmpty) {`
`731`	`731`	`output = output.withColumn($(linkPredictionCol), predictLinkUDF(col($(featuresCol))))`
`732`	`732`	`}`
`733`		`- output`
	`733`	`+ output.toDF`
`734`	`734`	`}`
`735`	`735`
`736`	`736`	`private var trainingSummary: Option[GeneralizedLinearRegressionSummary] = None`