Github user MrBago commented on a diff in the pull request:
https://github.com/apache/spark/pull/19843#discussion_r153964786
--- Diff:
mllib/src/test/scala/org/apache/spark/ml/regression/LinearRegressionSuite.scala
---
@@ -233,7 +232,8 @@ class LinearRegressionSuite
assert(model2.intercept ~== interceptR relTol 1E-3)
assert(model2.coefficients ~= coefficientsR relTol 1E-3)
- model1.transform(datasetWithDenseFeature).select("features",
"prediction").collect().foreach {
+ testTransformer[(Double, Vector)](datasetWithDenseFeature,
Seq(model1),
+ "features", "prediction") {
--- End diff --
I really like this, I'm impressed by how little we need to change to
leverage existing tests and have them run in both streaming and in batch mode!!
---
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
