donote / nmt-seq2seq

seq2seq model written in Pytorch

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

requirements:

  • python 3.6
  • pytorch 0.3.1
  • nltk
  • tqdm

How to use

  • To generate data set python train_test_split.py
  • To train the model, simply run ./run.sh
  • To test the model .run_test.sh
  • To see what config options you have python main.py --help

background

TODO

  • add attention to the currently very basic model

Bug report

If you find any bugs, please feel free to send an email to zeweichu@gmail.com or donote@163.com , I will try to be responsive!

About

seq2seq model written in Pytorch


Languages

Language:Python 97.1%Language:Shell 2.9%