public class BinaryClassificationEvaluator extends Evaluator implements Params, HasRawPredictionCol, HasLabelCol
Evaluator for binary classification, which expects two input columns: score and label.
Constructor and Description |
---|
BinaryClassificationEvaluator() |
Modifier and Type | Method and Description |
---|---|
double |
evaluate(DataFrame dataset,
ParamMap paramMap)
Evaluates the output.
|
String |
getMetricName() |
Param<String> |
metricName()
param for metric name in evaluation
|
BinaryClassificationEvaluator |
setLabelCol(String value) |
BinaryClassificationEvaluator |
setMetricName(String value) |
BinaryClassificationEvaluator |
setScoreCol(String value) |
equals, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait
getRawPredictionCol, rawPredictionCol
getLabelCol, labelCol
addOutputColumn, checkInputColumn, explainParams, get, getParam, isSet, paramMap, params, set, set, validate, validate
uid
public Param<String> metricName()
public String getMetricName()
public BinaryClassificationEvaluator setMetricName(String value)
public BinaryClassificationEvaluator setScoreCol(String value)
public BinaryClassificationEvaluator setLabelCol(String value)