Neural machine translation model using seq2seq, GRU GPU training, attention mechanism and subword segmentation.
Python: 3.8 Sklearn: 0.23.2 Tensorflow: 2.5.1 Keras: 2.2.5 Cuda: 11.1.1
Create virtual environment
Install prerequisite
Verify dataset
Setup path environment
Load checkpoint/weights
Run NMT.ipynb
Pull requests are welcome. For major changes, please open an issue first to discuss what you would like to change.
Please make sure to update tests as appropriate.