Repository: spark Updated Branches: refs/heads/master d0482f6af -> 3889b1c7a
[SPARK-11059] [ML] Change range of quantile probabilities in AFTSurvivalRegression Value of the quantile probabilities array should be in the range (0, 1) instead of [0,1] in `AFTSurvivalRegression.scala` according to [Discussion] (https://github.com/apache/spark/pull/8926#discussion-diff-40698242) Author: vectorijk <[email protected]> Closes #9083 from vectorijk/spark-11059. Project: http://git-wip-us.apache.org/repos/asf/spark/repo Commit: http://git-wip-us.apache.org/repos/asf/spark/commit/3889b1c7 Tree: http://git-wip-us.apache.org/repos/asf/spark/tree/3889b1c7 Diff: http://git-wip-us.apache.org/repos/asf/spark/diff/3889b1c7 Branch: refs/heads/master Commit: 3889b1c7a96da1111946fa63ad69489b83468646 Parents: d0482f6 Author: vectorijk <[email protected]> Authored: Tue Oct 13 15:57:36 2015 -0700 Committer: Xiangrui Meng <[email protected]> Committed: Tue Oct 13 15:57:36 2015 -0700 ---------------------------------------------------------------------- .../org/apache/spark/ml/regression/AFTSurvivalRegression.scala | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) ---------------------------------------------------------------------- http://git-wip-us.apache.org/repos/asf/spark/blob/3889b1c7/mllib/src/main/scala/org/apache/spark/ml/regression/AFTSurvivalRegression.scala ---------------------------------------------------------------------- diff --git a/mllib/src/main/scala/org/apache/spark/ml/regression/AFTSurvivalRegression.scala b/mllib/src/main/scala/org/apache/spark/ml/regression/AFTSurvivalRegression.scala index 717caac..ac2c3d8 100644 --- a/mllib/src/main/scala/org/apache/spark/ml/regression/AFTSurvivalRegression.scala +++ b/mllib/src/main/scala/org/apache/spark/ml/regression/AFTSurvivalRegression.scala @@ -59,14 +59,14 @@ private[regression] trait AFTSurvivalRegressionParams extends Params /** * Param for quantile probabilities array. - * Values of the quantile probabilities array should be in the range [0, 1] + * Values of the quantile probabilities array should be in the range (0, 1) * and the array should be non-empty. * @group param */ @Since("1.6.0") final val quantileProbabilities: DoubleArrayParam = new DoubleArrayParam(this, "quantileProbabilities", "quantile probabilities array", - (t: Array[Double]) => t.forall(ParamValidators.inRange(0, 1)) && t.length > 0) + (t: Array[Double]) => t.forall(ParamValidators.inRange(0, 1, false, false)) && t.length > 0) /** @group getParam */ @Since("1.6.0") --------------------------------------------------------------------- To unsubscribe, e-mail: [email protected] For additional commands, e-mail: [email protected]
