public static final class IdEmbedding.Builder
extends java.lang.Object
IdEmbedding type of Block.| Constructor and Description |
|---|
Builder() |
| Modifier and Type | Method and Description |
|---|---|
IdEmbedding |
build()
Builds the
IdEmbedding. |
IdEmbedding.Builder |
setDictionarySize(int dictionarySize)
Sets the number of ids that should be embedded.
|
IdEmbedding.Builder |
setEmbeddingSize(int embeddingSize)
Sets the size of the embeddings.
|
public IdEmbedding.Builder setDictionarySize(int dictionarySize)
dictionarySize - the number of ids that should be embedded. Valid ids are 0 to
dictionarySize - 1.public IdEmbedding.Builder setEmbeddingSize(int embeddingSize)
embeddingSize - the size of the embeddings.public IdEmbedding build()
IdEmbedding.IdEmbeddingjava.lang.IllegalArgumentException - if all required parameters (items, embeddingSize) have
not been set