Github user sethah commented on a diff in the pull request:
https://github.com/apache/spark/pull/19638#discussion_r148623948
--- Diff:
mllib/src/test/scala/org/apache/spark/ml/evaluation/RegressionEvaluatorSuite.scala
---
@@ -73,6 +73,11 @@ class RegressionEvaluatorSuite
evaluator.setMetricName("r2")
assert(evaluator.evaluate(predictions) ~== 0.9998387 absTol 0.01)
+ // Adjusted r2
+ evaluator.setMetricName("r2adj")
+ assert(evaluator.evaluate(predictions) ~== 0.9998 absTol 0.01)
+ print(evaluator.evaluate(predictions))
--- End diff --
remove
---
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
