Package com.basistech.rosette.dm

Class EmbeddingCollection.Builder

java.lang.Object

com.basistech.rosette.dm.EmbeddingCollection.Builder

Enclosing class:

EmbeddingCollection

public static class EmbeddingCollection.Builder
extends Object

Builder class for EmbeddingsCollection.

Constructor Summary

Constructors

Constructor	Description
Builder()
Builder(EmbeddingCollection toCopy)

Method Summary

Modifier and Type	Method	Description
EmbeddingCollection	build()	Build the collection.
Map<Integer,float[]>	embeddings()	Retrieve the current embeddings
EmbeddingCollection.Builder	put(int index, float[] values)	Add an embedding.

Methods inherited from class java.lang.Object

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

Constructor Details

Builder

public Builder()

Builder

public Builder(EmbeddingCollection toCopy)

Method Details

put

public EmbeddingCollection.Builder put(int index,
 float[] values)

Add an embedding.

Parameters:

index - the item index.

values - the values.

Returns:

this

embeddings

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

Retrieve the current embeddings

Returns:

the embeddings.

build

public EmbeddingCollection build()

Build the collection.

Returns:

the collection.