WordEmbedding.ConfigΒΆ

Component: WordEmbedding

class WordEmbedding.Config

Bases: ConfigBase

All Attributes (including base classes)

embed_dim: int = 100
freeze: bool = False
embedding_init_strategy: EmbedInitStrategy = <EmbedInitStrategy.RANDOM: 'random'>
embedding_init_range: Optional[List[float]] = None
export_input_names: List[str] = ['tokens_vals']
pretrained_embeddings_path: str = ''
vocab_file: str = ''
vocab_size: int = 0
vocab_from_train_data: bool = True
vocab_from_all_data: bool = False
vocab_from_pretrained_embeddings: bool = False
lowercase_tokens: bool = True
min_freq: int = 1
mlp_layer_dims: Optional[List[int]] = []

Default JSON

{
    "embed_dim": 100,
    "freeze": false,
    "embedding_init_strategy": "random",
    "embedding_init_range": null,
    "export_input_names": [
        "tokens_vals"
    ],
    "pretrained_embeddings_path": "",
    "vocab_file": "",
    "vocab_size": 0,
    "vocab_from_train_data": true,
    "vocab_from_all_data": false,
    "vocab_from_pretrained_embeddings": false,
    "lowercase_tokens": true,
    "min_freq": 1,
    "mlp_layer_dims": []
}