You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@spark.apache.org by yl...@apache.org on 2017/01/06 02:59:59 UTC
spark git commit: [MINOR] Correct LogisticRegression test case for
probability2prediction.
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 <yb...@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 <yb...@gmail.com>
Authored: Thu Jan 5 18:59:49 2017 -0800
Committer: Yanbo Liang <yb...@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-unsubscribe@spark.apache.org
For additional commands, e-mail: commits-help@spark.apache.org