diff options
-rw-r--r-- | mllib/src/test/scala/org/apache/spark/ml/clustering/LDASuite.scala | 3 | ||||
-rw-r--r-- | mllib/src/test/scala/org/apache/spark/ml/util/DefaultReadWriteTest.scala | 3 |
2 files changed, 5 insertions, 1 deletions
diff --git a/mllib/src/test/scala/org/apache/spark/ml/clustering/LDASuite.scala b/mllib/src/test/scala/org/apache/spark/ml/clustering/LDASuite.scala index 03270401ad..a3a8f65eac 100644 --- a/mllib/src/test/scala/org/apache/spark/ml/clustering/LDASuite.scala +++ b/mllib/src/test/scala/org/apache/spark/ml/clustering/LDASuite.scala @@ -52,7 +52,8 @@ object LDASuite { "checkpointInterval" -> 30, "learningOffset" -> 1023.0, "learningDecay" -> 0.52, - "subsamplingRate" -> 0.051 + "subsamplingRate" -> 0.051, + "docConcentration" -> Array(2.0) ) } diff --git a/mllib/src/test/scala/org/apache/spark/ml/util/DefaultReadWriteTest.scala b/mllib/src/test/scala/org/apache/spark/ml/util/DefaultReadWriteTest.scala index 0aa774b660..8e5365af84 100644 --- a/mllib/src/test/scala/org/apache/spark/ml/util/DefaultReadWriteTest.scala +++ b/mllib/src/test/scala/org/apache/spark/ml/util/DefaultReadWriteTest.scala @@ -82,6 +82,7 @@ trait DefaultReadWriteTest extends TempDirectory { self: Suite => * - Explicitly set Params, and train model * - Test save/load using [[testDefaultReadWrite()]] on Estimator and Model * - Check Params on Estimator and Model + * - Compare model data * * This requires that the [[Estimator]] and [[Model]] share the same set of [[Param]]s. * @param estimator Estimator to test @@ -117,6 +118,8 @@ trait DefaultReadWriteTest extends TempDirectory { self: Suite => val param = model.getParam(p) assert(model.get(param).get === model2.get(param).get) } + + checkModelData(model, model2) } } |