Repository: spark Updated Branches: refs/heads/master cca945b6a -> dfc4c935b
[MINOR] Correct LogisticRegression test case for probability2prediction. ## What changes were proposed in this pull request? Set correct column names for ```force to use probability2prediction``` in ```LogisticRegressionSuite```. ## How was this patch tested? Change unit test. Author: Yanbo Liang <yblia...@gmail.com> Closes #16477 from yanboliang/lor-pred. Project: http://git-wip-us.apache.org/repos/asf/spark/repo Commit: http://git-wip-us.apache.org/repos/asf/spark/commit/dfc4c935 Tree: http://git-wip-us.apache.org/repos/asf/spark/tree/dfc4c935 Diff: http://git-wip-us.apache.org/repos/asf/spark/diff/dfc4c935 Branch: refs/heads/master Commit: dfc4c935ba99a4adab2bf39cf24fff1ee3011759 Parents: cca945b Author: Yanbo Liang <yblia...@gmail.com> Authored: Thu Jan 5 18:59:49 2017 -0800 Committer: Yanbo Liang <yblia...@gmail.com> Committed: Thu Jan 5 18:59:49 2017 -0800 ---------------------------------------------------------------------- .../spark/ml/classification/LogisticRegressionSuite.scala | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) ---------------------------------------------------------------------- http://git-wip-us.apache.org/repos/asf/spark/blob/dfc4c935/mllib/src/test/scala/org/apache/spark/ml/classification/LogisticRegressionSuite.scala ---------------------------------------------------------------------- diff --git a/mllib/src/test/scala/org/apache/spark/ml/classification/LogisticRegressionSuite.scala b/mllib/src/test/scala/org/apache/spark/ml/classification/LogisticRegressionSuite.scala index 1308210..c14dcbd 100644 --- a/mllib/src/test/scala/org/apache/spark/ml/classification/LogisticRegressionSuite.scala +++ b/mllib/src/test/scala/org/apache/spark/ml/classification/LogisticRegressionSuite.scala @@ -360,7 +360,7 @@ class LogisticRegressionSuite } // force it to use raw2prediction - model.setProbabilityCol("") + model.setRawPredictionCol("rawPrediction").setProbabilityCol("") val resultsUsingRaw2Predict = model.transform(smallMultinomialDataset).select("prediction").as[Double].collect() resultsUsingRaw2Predict.zip(results.select("prediction").as[Double].collect()).foreach { @@ -368,7 +368,7 @@ class LogisticRegressionSuite } // force it to use probability2prediction - model.setRawPredictionCol("") + model.setRawPredictionCol("").setProbabilityCol("probability") val resultsUsingProb2Predict = model.transform(smallMultinomialDataset).select("prediction").as[Double].collect() resultsUsingProb2Predict.zip(results.select("prediction").as[Double].collect()).foreach { @@ -414,7 +414,7 @@ class LogisticRegressionSuite } // force it to use raw2prediction - model.setProbabilityCol("") + model.setRawPredictionCol("rawPrediction").setProbabilityCol("") val resultsUsingRaw2Predict = model.transform(smallBinaryDataset).select("prediction").as[Double].collect() resultsUsingRaw2Predict.zip(results.select("prediction").as[Double].collect()).foreach { @@ -422,7 +422,7 @@ class LogisticRegressionSuite } // force it to use probability2prediction - model.setRawPredictionCol("") + model.setRawPredictionCol("").setProbabilityCol("probability") val resultsUsingProb2Predict = model.transform(smallBinaryDataset).select("prediction").as[Double].collect() resultsUsingProb2Predict.zip(results.select("prediction").as[Double].collect()).foreach { --------------------------------------------------------------------- To unsubscribe, e-mail: commits-unsubscr...@spark.apache.org For additional commands, e-mail: commits-h...@spark.apache.org