https://github.com/minhnguyent546/nmt-en-vi

Neural machine translation English - Vietnamese with Transformer, trained from scratch
https://github.com/minhnguyent546/nmt-en-vi

natural-language-processing neural-machine-translation transformer

Last synced: 27 days ago
JSON representation

Neural machine translation English - Vietnamese with Transformer, trained from scratch

• Host: GitHub
• URL: https://github.com/minhnguyent546/nmt-en-vi
• Owner: minhnguyent546
• Created: 2024-02-11T13:50:49.000Z (11 months ago)
• Default Branch: master
• Last Pushed: 2024-05-19T07:46:26.000Z (8 months ago)
• Last Synced: 2024-05-19T14:53:17.982Z (8 months ago)
• Topics: natural-language-processing, neural-machine-translation, transformer
• Language: Python
• Homepage:
• Size: 4.23 MB
• Stars: 0
• Watchers: 2
• Forks: 0
• Open Issues: 0
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

README

        
# Neural machine translation English-Vietnamese with Transformer



  





  



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:

```bash

git clone https://github.com/minhnguyent546/nmt-en-vi.git

cd nmt-en-vi

```

- Install required dependencies:

```bash

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:

```bash

python preprocess_nmt.py --config 'config/config.yaml'

```

- To train the model:

```bash

python train_nmt.py --config 'config/config.yaml'

```

- To test the model:

```bash

python test_nmt.py --config 'config/config.yaml'

```