diff options
| author | Yu ISHIKAWA <yuu.ishikawa@gmail.com> | 2016-02-11 15:05:34 -0800 |
|---|---|---|
| committer | Xiangrui Meng <meng@databricks.com> | 2016-02-11 15:05:34 -0800 |
| commit | 574571c87098795a2206a113ee9ed4bafba8f00f (patch) | |
| tree | 6552478f8e19aecba3fe13b026ec85ddafaa6966 /mllib/src/test/scala | |
| parent | efb65e09bcfa4542348f5cd37fe5c14047b862e5 (diff) | |
| download | spark-574571c87098795a2206a113ee9ed4bafba8f00f.tar.gz spark-574571c87098795a2206a113ee9ed4bafba8f00f.tar.bz2 spark-574571c87098795a2206a113ee9ed4bafba8f00f.zip | |
[SPARK-11515][ML] QuantileDiscretizer should take random seed
cc jkbradley
Author: Yu ISHIKAWA <yuu.ishikawa@gmail.com>
Closes #9535 from yu-iskw/SPARK-11515.
Diffstat (limited to 'mllib/src/test/scala')
| -rw-r--r-- | mllib/src/test/scala/org/apache/spark/ml/feature/QuantileDiscretizerSuite.scala | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/mllib/src/test/scala/org/apache/spark/ml/feature/QuantileDiscretizerSuite.scala b/mllib/src/test/scala/org/apache/spark/ml/feature/QuantileDiscretizerSuite.scala index 722f1abde4..4fde42972f 100644 --- a/mllib/src/test/scala/org/apache/spark/ml/feature/QuantileDiscretizerSuite.scala +++ b/mllib/src/test/scala/org/apache/spark/ml/feature/QuantileDiscretizerSuite.scala @@ -93,7 +93,7 @@ private object QuantileDiscretizerSuite extends SparkFunSuite { val df = sc.parallelize(data.map(Tuple1.apply)).toDF("input") val discretizer = new QuantileDiscretizer().setInputCol("input").setOutputCol("result") - .setNumBuckets(numBucket) + .setNumBuckets(numBucket).setSeed(1) val result = discretizer.fit(df).transform(df) val transformedFeatures = result.select("result").collect() |