Package com.basistech.rosette.dm

Class EmbeddingCollection

java.lang.Object

com.basistech.rosette.dm.EmbeddingCollection

All Implemented Interfaces:

Serializable

public class EmbeddingCollection
extends Object
implements Serializable

A vector of embeddings for some vector of items in an AnnotatedText. Embeddings can exist for the entire text and for some or all the sub-items, such as tokens or entities. The collection of embeddings is represented as a map from indices to the value vector. In the case of the embedding for the entire text, the index value is 0.

See Also:

• Serialized Form

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static class	EmbeddingCollection.Builder	Builder class for EmbeddingsCollection.

Constructor Summary

Constructors

Modifier	Constructor	Description
protected	EmbeddingCollection(Map<Integer,float[]> embeddings)

Method Summary

Modifier and Type	Method	Description
boolean	equals(Object o)
Map<Integer,float[]>	getEmbeddings()	Return the embeddings.
int	hashCode()
String	toString()

Methods inherited from class java.lang.Object

clone, finalize, getClass, notify, notifyAll, wait, wait, wait

Constructor Details

EmbeddingCollection

protected EmbeddingCollection(Map<Integer,float[]> embeddings)

Method Details

getEmbeddings

public Map<Integer,float[]> getEmbeddings()

Return the embeddings.

Returns:

a map from item indices to values.

equals

public boolean equals(Object o)

Overrides:

equals in class Object

hashCode

public int hashCode()

Overrides:

hashCode in class Object

toString

public String toString()

Overrides:

toString in class Object