Package ai.djl.modality.nlp

Class EmbeddingOutput

java.lang.Object

ai.djl.modality.nlp.EmbeddingOutput

public class EmbeddingOutput
extends Object

A class represents text embedding output.

Constructor Summary

Constructors

Constructor	Description
EmbeddingOutput()

Method Summary

Modifier and Type	Method	Description
void	addTokenWeights(String tokenId, float weights)	Adds the token weights to the output.
float[]	getDenseEmbedding()	Returns the dense embedding.
Map<String,Float>	getLexicalWeights()	Returns the token weights map.
void	setDenseEmbedding(float[] denseEmbedding)	Sets the dense embedding.

Methods inherited from class java.lang.Object

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

Constructor Details

EmbeddingOutput

public EmbeddingOutput()

Method Details

getDenseEmbedding

public float[] getDenseEmbedding()

Returns the dense embedding.

Returns:

the dense embedding

setDenseEmbedding

public void setDenseEmbedding(float[] denseEmbedding)

Sets the dense embedding.

Parameters:

denseEmbedding - the dense embedding

getLexicalWeights

public Map<String,Float> getLexicalWeights()

Returns the token weights map.

Returns:

the token weights map

addTokenWeights

public void addTokenWeights(String tokenId,
 float weights)

Adds the token weights to the output.

Parameters:

tokenId - the token id

weights - the token weights