Package ai.djl.modality

Class Classifications.Classification

java.lang.Object

ai.djl.modality.Classifications.Classification

Direct Known Subclasses:

DetectedObjects.DetectedObject

Enclosing class:

Classifications

public static class Classifications.Classification
extends Object

A Classification stores the classification result for a single class on a single input.

Constructor Summary

Constructors

Constructor	Description
Classification(String className, double probability)	Constructs a single class result for a classification.

Method Summary

Modifier and Type	Method	Description
String	getClassName()	Returns the class name.
double	getProbability()	Returns the probability.
String	toString()

Methods inherited from class java.lang.Object

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

Constructor Details

Classification

public Classification(String className,
 double probability)

Constructs a single class result for a classification.

Parameters:

className - the class name of the result

probability - the probability of the result

Method Details

getClassName

public String getClassName()

Returns the class name.

Returns:

the class name

getProbability

public double getProbability()

Returns the probability.

Probability explains how accurately the classifier identified the target class.

Returns:

the probability

toString

public String toString()

Overrides:

toString in class Object