/seq2seq

reverse fixed-length texts

Primary LanguagePython

seq2seq

description

In this repo, I have written and refactored some code to mirror strings using the seq2seq model. For example, if the input is ABC, the ground-truth output is CBA.

references

requirements

CUDA 9.0 + Pytorch 1.0

to run

ROOT = path/to/this/repo

  • download the dataset
cd ROOT
python dataset/download.py
  • training:
python main.py
  • inference:
python inference.py

future works

attention, word embedding (glove or bert)