Skip to content

Commit

Permalink
[SPARK-14618][ML][DOC] Updated RegressionEvaluator.metricName param doc
Browse files Browse the repository at this point in the history
## What changes were proposed in this pull request?

In Spark 1.4, we negated some metrics from RegressionEvaluator since CrossValidator always maximized metrics. This was fixed in 1.5, but the docs were not updated. This PR updates the docs.

## How was this patch tested?

no tests

Author: Joseph K. Bradley <[email protected]>

Closes apache#12377 from jkbradley/regeval-doc.
  • Loading branch information
jkbradley committed Apr 14, 2016
1 parent c5172f8 commit bf65c87
Showing 1 changed file with 5 additions and 4 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -39,11 +39,12 @@ final class RegressionEvaluator @Since("1.4.0") (@Since("1.4.0") override val ui
def this() = this(Identifiable.randomUID("regEval"))

/**
* param for metric name in evaluation (supports `"rmse"` (default), `"mse"`, `"r2"`, and `"mae"`)
* Param for metric name in evaluation. Supports:
* - `"rmse"` (default): root mean squared error
* - `"mse"`: mean squared error
* - `"r2"`: R^2^ metric
* - `"mae"`: mean absolute error
*
* Because we will maximize evaluation value (ref: `CrossValidator`),
* when we evaluate a metric that is needed to minimize (e.g., `"rmse"`, `"mse"`, `"mae"`),
* we take and output the negative of this metric.
* @group param
*/
@Since("1.4.0")
Expand Down

0 comments on commit bf65c87

Please sign in to comment.