RNN moduleΒΆ
Defines a nn.Module from pytorch where we define the architecture for the Recurrent Neural Network used for the Recommedation System.
The archicture consititutes of Embedding Layer of size 64 * size of input words. This is followed by an LSTM layer of 64(embedding size) * 64(lstm size). Finally a Dense Linear Layer is used to give the output for the model.
RNNModule(
(embedding): Embedding(91, 64)
(lstm): LSTM(64, 64, batch_first=True)
(dense): Linear(in_features=64, out_features=91, bias=True)
)