BiLSTMDocSlotAttention.ConfigΒΆ

Component: BiLSTMDocSlotAttention

class BiLSTMDocSlotAttention.Config[source]

Bases: RepresentationBase.Config, ConfigBase

All Attributes (including base classes)

load_path: Optional[str] = None
save_path: Optional[str] = None
freeze: bool = False
shared_module_key: Optional[str] = None
dropout: float = 0.4
lstm: Union[BiLSTM.Config, OrderedNeuronLSTM.Config, AugmentedLSTM.Config] = BiLSTM.Config()
pooling: Union[SelfAttention.Config, MaxPool.Config, MeanPool.Config, NoneType] = None
slot_attention: Optional[SlotAttention.Config] = None
doc_mlp_layers: int = 0
word_mlp_layers: int = 0

Default JSON

{
    "load_path": null,
    "save_path": null,
    "freeze": false,
    "shared_module_key": null,
    "dropout": 0.4,
    "lstm": {
        "BiLSTM": {
            "load_path": null,
            "save_path": null,
            "freeze": false,
            "shared_module_key": null,
            "dropout": 0.4,
            "lstm_dim": 32,
            "num_layers": 1,
            "bidirectional": true,
            "pack_sequence": true,
            "disable_sort_in_jit": false
        }
    },
    "pooling": null,
    "slot_attention": null,
    "doc_mlp_layers": 0,
    "word_mlp_layers": 0
}