DictEmbedding.ConfigΒΆ

Component: DictEmbedding

class DictEmbedding.Config

Bases: ConfigBase

All Attributes (including base classes)

embed_dim: int = 100
sparse: bool = False
pooling: PoolingType = <PoolingType.MEAN: 'mean'>
export_input_names: List[str] = ['dict_vals', 'dict_weights', 'dict_lens']
vocab_from_train_data: bool = True

Default JSON

{
    "embed_dim": 100,
    "sparse": false,
    "pooling": "mean",
    "export_input_names": [
        "dict_vals",
        "dict_weights",
        "dict_lens"
    ],
    "vocab_from_train_data": true
}