Skip to content

Neural machine translation English - Vietnamese with Transformer.

Notifications You must be signed in to change notification settings

minhnguyent546/nmt-en-vi

Repository files navigation

Neural machine translation English-Vietnamese with Transformer

Open in Colab Open in Kaggle

A Transformer implementation from scratch, crafted following the principles outlined in the paper 'Attention Is All You Need', serves as a robust solution for tasks such as neural machine translation and grammatical error correction.

Setup

  • Clone this repo:
git clone https://github.com/minhnguyent546/nmt-en-vi.git
cd nmt-en-vi
  • Install required dependencies:
pip install -r requirements.txt

Training and testing

Please take a look at the config file located at config/config.yaml, and change the train, test, and validation paths to your local files.

  • Preprocessing the data:
python preprocess_nmt.py --config 'config/config.yaml'
  • To train the model:
python train_nmt.py --config 'config/config.yaml'
  • To test the model:
python test_nmt.py --config 'config/config.yaml'

About

Neural machine translation English - Vietnamese with Transformer.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published