WordEmbedding.Config

Component: WordEmbedding

class WordEmbedding.Config

Bases: Module.Config

All Attributes (including base classes)

load_path: Optional[str] = None

save_path: Optional[str] = None

freeze: bool = False

shared_module_key: Optional[str] = None

embed_dim: int = 100

embedding_init_strategy: EmbedInitStrategy = <EmbedInitStrategy.RANDOM: 'random'>

embedding_init_range: Optional[list[float]] = None

embeddding_init_std: Optional[float] = 0.02

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]] = []

padding_idx: Optional[int] = None

cpu_only: bool = False

skip_header: bool = True

delimiter: str = ' '

Default JSON

{
    "load_path": null,
    "save_path": null,
    "freeze": false,
    "shared_module_key": null,
    "embed_dim": 100,
    "embedding_init_strategy": "random",
    "embedding_init_range": null,
    "embeddding_init_std": 0.02,
    "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": [],
    "padding_idx": null,
    "cpu_only": false,
    "skip_header": true,
    "delimiter": " "
}