public class MulticlassClassificationEvaluator extends Evaluator implements HasPredictionCol, HasLabelCol, DefaultParamsWritable
Constructor and Description |
---|
MulticlassClassificationEvaluator() |
MulticlassClassificationEvaluator(String uid) |
Modifier and Type | Method and Description |
---|---|
MulticlassClassificationEvaluator |
copy(ParamMap extra)
Creates a copy of this instance with the same UID and some extra params.
|
double |
evaluate(Dataset<?> dataset)
Evaluates model output and returns a scalar metric.
|
String |
getMetricName() |
boolean |
isLargerBetter()
Indicates whether the metric returned by
evaluate should be maximized (true, default)
or minimized (false). |
static MulticlassClassificationEvaluator |
load(String path) |
Param<String> |
metricName()
param for metric name in evaluation (supports
"f1" (default), "weightedPrecision" ,
"weightedRecall" , "accuracy" ) |
static MLReader<T> |
read() |
MulticlassClassificationEvaluator |
setLabelCol(String value) |
MulticlassClassificationEvaluator |
setMetricName(String value) |
MulticlassClassificationEvaluator |
setPredictionCol(String value) |
String |
uid()
An immutable unique ID for the object and its derivatives.
|
equals, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait
getPredictionCol, predictionCol
getLabelCol, labelCol
clear, copyValues, defaultCopy, defaultParamMap, explainParam, explainParams, extractParamMap, extractParamMap, get, getDefault, getOrDefault, getParam, hasDefault, hasParam, isDefined, isSet, paramMap, params, set, set, set, setDefault, setDefault, shouldOwn
toString
write
save
public MulticlassClassificationEvaluator(String uid)
public MulticlassClassificationEvaluator()
public static MulticlassClassificationEvaluator load(String path)
public static MLReader<T> read()
public String uid()
Identifiable
uid
in interface Identifiable
public Param<String> metricName()
"f1"
(default), "weightedPrecision"
,
"weightedRecall"
, "accuracy"
)public String getMetricName()
public MulticlassClassificationEvaluator setMetricName(String value)
public MulticlassClassificationEvaluator setPredictionCol(String value)
public MulticlassClassificationEvaluator setLabelCol(String value)
public double evaluate(Dataset<?> dataset)
Evaluator
isLargerBetter
specifies whether larger values are better.
public boolean isLargerBetter()
Evaluator
evaluate
should be maximized (true, default)
or minimized (false).
A given evaluator may support multiple metrics which may be maximized or minimized.isLargerBetter
in class Evaluator
public MulticlassClassificationEvaluator copy(ParamMap extra)
Params
defaultCopy()
.