Github user yanboliang commented on a diff in the pull request:
https://github.com/apache/spark/pull/19525#discussion_r155414954
--- Diff:
mllib/src/test/scala/org/apache/spark/ml/classification/LogisticRegressionSuite.scala
---
@@ -2769,6 +2769,20 @@ class LogisticRegressionSuite
LogisticRegressionSuite.allParamSettings, checkModelData)
}
+ test("read/write with BoundsOnCoefficients") {
+ def checkModelData(model: LogisticRegressionModel, model2:
LogisticRegressionModel): Unit = {
+ assert(model.getLowerBoundsOnCoefficients ===
model2.getLowerBoundsOnCoefficients)
+ assert(model.getUpperBoundsOnCoefficients ===
model2.getUpperBoundsOnCoefficients)
--- End diff --
Sure, you can update existing read/write test or your test, as long as to
check model itself rather than parameters.
---
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
