/sequential_model_pytorch_rnn_1dcnn

Sequential data using RNN (LSTM, GRU, etc) CNN and RCNN. Adaptable to variable length input sequence.

Primary LanguagePython

Sequential Model Pytorch

The repo provides library supporting:

  • RNN, LSTM, GRU, bi-GRU, 1d-CNN, RCNN, etc.
  • Adaptable to variable length input sequence.

Quick start

Choose the type of neural networks in train.py

To train:

python train.py

To test:

python test.py