RandomLSTM
- class lightautoml.text.dl_transformers.RandomLSTM(embed_size=300, hidden_size=256, pooling='mean', num_layers=1, **kwargs)[source]
Bases:
Module
Class to compute Random LSTM sentence embeddings from words embeddings.
- Parameters:
Note
There are several pooling types:
‘max’: Maximum on seq_len dimension for non masked inputs.
‘mean’: Mean on seq_len dimension for non masked inputs.
‘sum’: Sum on seq_len dimension for non masked inputs.