public final class RegressionEvaluator extends Evaluator
Constructor and Description |
---|
RegressionEvaluator() |
RegressionEvaluator(String uid) |
Modifier and Type | Method and Description |
---|---|
double |
evaluate(DataFrame dataset)
Evaluates the output.
|
String |
getMetricName() |
Param<String> |
metricName()
param for metric name in evaluation (supports
"rmse" (default), "mse" , "r2" , and "mae" ) |
RegressionEvaluator |
setLabelCol(String value) |
RegressionEvaluator |
setMetricName(String value) |
RegressionEvaluator |
setPredictionCol(String value) |
String |
uid() |
equals, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait
clear, copyValues, defaultParamMap, explainParam, explainParams, extractParamMap, extractParamMap, get, getDefault, getOrDefault, getParam, hasDefault, hasParam, isDefined, isSet, paramMap, params, set, set, set, setDefault, setDefault, setDefault, shouldOwn, validateParams
public RegressionEvaluator(String uid)
public RegressionEvaluator()
public String uid()
public Param<String> metricName()
"rmse"
(default), "mse"
, "r2"
, and "mae"
)public String getMetricName()
public RegressionEvaluator setMetricName(String value)
public RegressionEvaluator setPredictionCol(String value)
public RegressionEvaluator setLabelCol(String value)