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, index

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

totalInstances

Constructor Summary

Constructors

Constructor and Description
Accuracy() Creates a multiclass accuracy evaluator that computes accuracy across axis 1 along the 0th index.
Accuracy(java.lang.String name, int index) Creates a multiclass accuracy evaluator that computes accuracy across axis 1 along given index.
Accuracy(java.lang.String name, int index, int axis) Creates a multiclass accuracy evaluator.

Method Summary

Modifier and Type	Method and Description
protected ai.djl.util.Pair<java.lang.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

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 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,
                int index)

Creates a multiclass accuracy evaluator that computes accuracy across axis 1 along given index.

Parameters:

name - the name of the evaluator, default is "Accuracy"

index - the index of the NDArray in labels to compute accuracy for

Accuracy

public Accuracy(java.lang.String name,
                int index,
                int axis)

Creates a multiclass accuracy evaluator.

Parameters:

name - the name of the evaluator, default is "Accuracy"

index - the index of the NDArray in labels to compute accuracy for

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 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)