-
- Downloads
[ML] testEstimatorAndModelReadWrite should call checkModelData
## What changes were proposed in this pull request? Although we defined ```checkModelData``` in [```read/write``` test](https://github.com/apache/spark/blob/master/mllib/src/test/scala/org/apache/spark/ml/regression/LinearRegressionSuite.scala#L994) of ML estimators/models and pass it to ```testEstimatorAndModelReadWrite```, ```testEstimatorAndModelReadWrite``` omits to call ```checkModelData``` to check the equality of model data. So actually we did not run the check of model data equality for all test cases currently, we should fix it. BTW, fix the bug of LDA read/write test which did not set ```docConcentration```. This bug should have failed test, but it does not complain because we did not run ```checkModelData``` actually. cc jkbradley mengxr ## How was this patch tested? No new unit test, should pass the exist ones. Author: Yanbo Liang <ybliang8@gmail.com> Closes #11513 from yanboliang/ml-check-model-data.
Showing
- mllib/src/test/scala/org/apache/spark/ml/clustering/LDASuite.scala 2 additions, 1 deletion.../test/scala/org/apache/spark/ml/clustering/LDASuite.scala
- mllib/src/test/scala/org/apache/spark/ml/util/DefaultReadWriteTest.scala 3 additions, 0 deletions...scala/org/apache/spark/ml/util/DefaultReadWriteTest.scala
Loading
Please register or sign in to comment