LSTM model
rnn_model(input_size, hidden_size, output_size, num_layers = 2, dropout = 0.3)Arguments
- input_size
input size (nr of columns with time series drivers)
- hidden_size
the size of the LSTM hidden layers in the module
- output_size
the output size of the model
- num_layers
number of layers in the LSTM module
- dropout
dropout rate of the LSTM module
Value
a bare LSTM model, to be configured and run