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 03270401ad2bbbef80f44befba3ab1dbd988b1ea..a3a8f65eac176c388d3350539b8c2f87ef637d11 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 0aa774b66078ed0685c9c94520c31f6cc6cd4929..8e5365af849a7e593c5f23f4364158fe3c06e9be 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) } }