diff options
author | MechCoder <manojkumarsivaraj334@gmail.com> | 2015-05-29 11:36:41 -0700 |
---|---|---|
committer | Xiangrui Meng <meng@databricks.com> | 2015-05-29 11:36:48 -0700 |
commit | 4be701aa50711ff22a0a0d7a87c9857ea2ef8e22 (patch) | |
tree | 69924a9758981e5f24486b815f5a44a588185fd3 /mllib/src/test | |
parent | 645e611644be3f62ef07e4ca7628bf298349d9a6 (diff) | |
download | spark-4be701aa50711ff22a0a0d7a87c9857ea2ef8e22.tar.gz spark-4be701aa50711ff22a0a0d7a87c9857ea2ef8e22.tar.bz2 spark-4be701aa50711ff22a0a0d7a87c9857ea2ef8e22.zip |
[SPARK-7946] [MLLIB] DecayFactor wrongly set in StreamingKMeans
Author: MechCoder <manojkumarsivaraj334@gmail.com>
Closes #6497 from MechCoder/spark-7946 and squashes the following commits:
2fdd0a3 [MechCoder] Add non-regression test
8c988c6 [MechCoder] [SPARK-7946] DecayFactor wrongly set in StreamingKMeans
(cherry picked from commit 6181937f315480543d28e542d43269cfa591e9d0)
Signed-off-by: Xiangrui Meng <meng@databricks.com>
Diffstat (limited to 'mllib/src/test')
-rw-r--r-- | mllib/src/test/scala/org/apache/spark/mllib/clustering/StreamingKMeansSuite.scala | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/mllib/src/test/scala/org/apache/spark/mllib/clustering/StreamingKMeansSuite.scala b/mllib/src/test/scala/org/apache/spark/mllib/clustering/StreamingKMeansSuite.scala index f90025d535..13f9b17c02 100644 --- a/mllib/src/test/scala/org/apache/spark/mllib/clustering/StreamingKMeansSuite.scala +++ b/mllib/src/test/scala/org/apache/spark/mllib/clustering/StreamingKMeansSuite.scala @@ -133,6 +133,13 @@ class StreamingKMeansSuite extends FunSuite with TestSuiteBase { assert(math.abs(c1) ~== 0.8 absTol 0.6) } + test("SPARK-7946 setDecayFactor") { + val kMeans = new StreamingKMeans() + assert(kMeans.decayFactor === 1.0) + kMeans.setDecayFactor(2.0) + assert(kMeans.decayFactor === 2.0) + } + def StreamingKMeansDataGenerator( numPoints: Int, numBatches: Int, |