BiLSTMDocAttention.Config¶
Component: BiLSTMDocAttention
-
class
BiLSTMDocAttention.
Config
[source] Bases:
RepresentationBase.Config
Configuration class for BiLSTM.
-
dropout
¶ Dropout probability to use. Defaults to 0.4.
Type: float
-
lstm
¶ Config for the BiLSTM.
Type: BiLSTM.Config
-
pooling
¶ Config for the underlying pooling module.
Type: ConfigBase
-
mlp_decoder
¶ Config for the non-linear projection module.
Type: MLPDecoder.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
- dropout: float =
0.4
- lstm: BiLSTM.Config = BiLSTM.Config()
- pooling: Union[SelfAttention.Config, MaxPool.Config, MeanPool.Config, NoPool.Config, LastTimestepPool.Config] = SelfAttention.Config()
- mlp_decoder: Optional[MLPDecoder.Config] =
None
Default JSON
{
"load_path": null,
"save_path": null,
"freeze": false,
"shared_module_key": null,
"dropout": 0.4,
"lstm": {
"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": {
"SelfAttention": {
"attn_dimension": 64,
"dropout": 0.4
}
},
"mlp_decoder": null
}