Package ai.djl.training.evaluator
Class Accuracy
- java.lang.Object
-
- ai.djl.training.evaluator.Evaluator
-
- ai.djl.training.evaluator.AbstractAccuracy
-
- ai.djl.training.evaluator.Accuracy
-
public class Accuracy extends AbstractAccuracy
Accuracy
is theAbstractAccuracy
with multiple classes.
-
-
Field Summary
-
Fields inherited from class ai.djl.training.evaluator.AbstractAccuracy
axis, correctInstances
-
Fields inherited from class ai.djl.training.evaluator.Evaluator
totalInstances
-
-
Constructor Summary
Constructors Constructor Description Accuracy()
Creates a multiclass accuracy evaluator that computes accuracy across axis 1 along the 0th index.Accuracy(java.lang.String name)
Creates a multiclass accuracy evaluator that computes accuracy across axis 1 along the 0th index.Accuracy(java.lang.String name, int axis)
Creates a multiclass accuracy evaluator.
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description protected ai.djl.util.Pair<java.lang.Long,NDArray>
accuracyHelper(NDList labels, NDList predictions)
A helper for classes extendingAbstractAccuracy
.-
Methods inherited from class ai.djl.training.evaluator.AbstractAccuracy
addAccumulator, evaluate, getAccumulator, resetAccumulator, updateAccumulator
-
Methods inherited from class ai.djl.training.evaluator.Evaluator
checkLabelShapes, checkLabelShapes, getName
-
-
-
-
Constructor Detail
-
Accuracy
public Accuracy()
Creates a multiclass accuracy evaluator that computes accuracy across axis 1 along the 0th index.
-
Accuracy
public Accuracy(java.lang.String name)
Creates a multiclass accuracy evaluator that computes accuracy across axis 1 along the 0th index.- Parameters:
name
- the name of the evaluator, default is "Accuracy"
-
Accuracy
public Accuracy(java.lang.String name, int axis)
Creates a multiclass accuracy evaluator.- Parameters:
name
- the name of the evaluator, default is "Accuracy"axis
- the axis that represent classes in prediction, default 1
-
-
Method Detail
-
accuracyHelper
protected ai.djl.util.Pair<java.lang.Long,NDArray> accuracyHelper(NDList labels, NDList predictions)
A helper for classes extendingAbstractAccuracy
.- Specified by:
accuracyHelper
in classAbstractAccuracy
- Parameters:
labels
- the labels to get accuracy forpredictions
- the predictions to get accuracy for- Returns:
- a pair(number of total values, ndarray int of correct values)
-
-