/chr_en_NMT

A Neural Machine Translation implementation in PyTorch that also serves as a template for building Seq2Seq models.

Primary LanguagePython

chr_en_NMT

A Neural Machine Translation implementation in PyTorch that also serves as a template for building Seq2Seq models.

Project Structure

./
├── checkpoint
├── chr_en_data
│   ├── dev.chr
│   ├── dev.en
│   ├── test.chr
│   ├── test.en
│   ├── train.chr
│   └── train.en
├── dataset
│   ├── dataset.py
├── decode.py
├── models
│   ├── lstm_seq2seq.py
├── pytest.ini
├── README.md
├── results
├── run.py
├── scripts
│   ├── decode.sh
│   ├── generate_vocab.sh
│   └── run.sh
├── src.model
├── src.vocab
├── tests
│   ├── __init__.py
│   ├── test_bleu.py
│   ├── test_dataset.py
│   ├── test.ipynb
│   ├── test_model.py
│   └── test_vocabEntry.py
├── tgt.model
├── tgt.vocab
├── utils.py
├── vocab.json
└── vocab.py

Usage

  1. generate sentence piece model
./scripts/generate_vocab.sh
  1. train
./scripts/run.sh
  1. decode
./scripts/decode.sh

Reference