Package ai.djl.modality.nlp.translator

Class ZeroShotClassificationOutput

java.lang.Object

ai.djl.modality.nlp.translator.ZeroShotClassificationOutput

public class ZeroShotClassificationOutput
extends Object

A class that represents a ZeroShotClassificationOutput object.

Constructor Summary

Constructors

Constructor	Description
ZeroShotClassificationOutput(String sequence, String[] labels, double[] scores)	Constructs a new ZeroShotClassificationOutput instance.

Method Summary

Modifier and Type	Method	Description
String[]	getLabels()	Returns the labels in sorted order.
double[]	getScores()	Returns the scores of the labels.
String	getSequence()	Returns the input text.

Methods inherited from class java.lang.Object

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

Constructor Details

ZeroShotClassificationOutput

public ZeroShotClassificationOutput(String sequence,
 String[] labels,
 double[] scores)

Constructs a new ZeroShotClassificationOutput instance.

Parameters:

sequence - the input text

labels - the labels

scores - the scores of the labels

Method Details

getSequence

public String getSequence()

Returns the input text.

Returns:

the input text

getLabels

public String[] getLabels()

Returns the labels in sorted order.

Returns:

the labels in sorted order

getScores

public double[] getScores()

Returns the scores of the labels.

Returns:

the scores of the labels