get_rnn#

pytorch_forecasting.models.nn.rnn.get_rnn(cell_type: Union[Type[RNN], str]) Type[RNN][source]#

Get LSTM or GRU.

Parameters

cell_type (Union[RNN, str]) – “LSTM” or “GRU”

Returns

returns GRU or LSTM RNN module

Return type

Type[RNN]