oracle.pgx.api.mllib

Class EmbeddingTableConfigBuilder

java.lang.Object

oracle.pgx.api.mllib.CategoricalPropertyConfigBuilder<EmbeddingTableConfig,EmbeddingTableConfigBuilder>

oracle.pgx.api.mllib.EmbeddingTableConfigBuilder

public class EmbeddingTableConfigBuilder
extends CategoricalPropertyConfigBuilder<EmbeddingTableConfig,EmbeddingTableConfigBuilder>

Builder for EmbeddingTableConfig.

Since:

23.2

Method Summary

Modifier and Type	Method and Description
EmbeddingTableConfig	build() Build the configuration
EmbeddingTableConfigBuilder	setEmbeddingDimension(int dim) Set the dimension of the vectors that categories will be encoded to in the embedding table.
EmbeddingTableConfigBuilder	setOutOfVocabularyProbability(double outOfVocabularyProbability) Set the probability to set category values in the input data to the out of vocabulary token randomly during training to learn a meaningful OOV embedding.

Methods inherited from class oracle.pgx.api.mllib.CategoricalPropertyConfigBuilder

embeddingTable, oneHotEncoding, setMaxVocabularySize, setShared

Methods inherited from class java.lang.Object

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

Method Detail

build

public EmbeddingTableConfig build()

Build the configuration

Overrides:

build in class CategoricalPropertyConfigBuilder<EmbeddingTableConfig,EmbeddingTableConfigBuilder>

Returns:

built configuration

Since:

23.2

setEmbeddingDimension

public EmbeddingTableConfigBuilder setEmbeddingDimension(int dim)

Set the dimension of the vectors that categories will be encoded to in the embedding table.

Parameters:

dim - embedding dimension

Returns:

this

Since:

23.2

setOutOfVocabularyProbability

public EmbeddingTableConfigBuilder setOutOfVocabularyProbability(double outOfVocabularyProbability)

Set the probability to set category values in the input data to the out of vocabulary token randomly during training to learn a meaningful OOV embedding. This procedure is disabled during inference.

Parameters:

outOfVocabularyProbability - probability of randomly setting a token to the out of vocabulary token

Returns:

this

Since:

23.2