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 the AbstractAccuracy 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(String name)	Creates a multiclass accuracy evaluator that computes accuracy across axis 1 along the 0th index.
Accuracy(String name, int axis)	Creates a multiclass accuracy evaluator.

Method Summary

Modifier and Type	Method	Description
protected ai.djl.util.Pair<Long,NDArray>	accuracyHelper(NDList labels, NDList predictions)	A helper for classes extending AbstractAccuracy.

Methods inherited from class ai.djl.training.evaluator.AbstractAccuracy

addAccumulator, evaluate, getAccumulator, resetAccumulator, updateAccumulator, updateAccumulators

Methods inherited from class ai.djl.training.evaluator.Evaluator

checkLabelShapes, checkLabelShapes, getName

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Details

Accuracy

public Accuracy()

Creates a multiclass accuracy evaluator that computes accuracy across axis 1 along the 0th index.

Accuracy

public Accuracy(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(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 Details

accuracyHelper

protected ai.djl.util.Pair<Long,NDArray> accuracyHelper(NDList labels,
 NDList predictions)

A helper for classes extending AbstractAccuracy.

Specified by:

accuracyHelper in class AbstractAccuracy

Parameters:

labels - the labels to get accuracy for

predictions - the predictions to get accuracy for

Returns:

a pair(number of total values, ndarray int of correct values)