diff options
author | vectorijk <jiangkai@gmail.com> | 2015-10-13 15:57:36 -0700 |
---|---|---|
committer | Xiangrui Meng <meng@databricks.com> | 2015-10-13 15:57:36 -0700 |
commit | 3889b1c7a96da1111946fa63ad69489b83468646 (patch) | |
tree | 928513bfa4c6da1a704794c63a6e706b98c57ac4 /mllib | |
parent | d0482f6af33e976db237405b2a978db1b7c2fd5b (diff) | |
download | spark-3889b1c7a96da1111946fa63ad69489b83468646.tar.gz spark-3889b1c7a96da1111946fa63ad69489b83468646.tar.bz2 spark-3889b1c7a96da1111946fa63ad69489b83468646.zip |
[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 <jiangkai@gmail.com>
Closes #9083 from vectorijk/spark-11059.
Diffstat (limited to 'mllib')
-rw-r--r-- | mllib/src/main/scala/org/apache/spark/ml/regression/AFTSurvivalRegression.scala | 4 |
1 files changed, 2 insertions, 2 deletions
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 717caacad3..ac2c3d825f 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") |