Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
cornerCao authored Apr 8, 2018
1 parent e8b9be1 commit 24d494a
Showing 1 changed file with 5 additions and 5 deletions.
10 changes: 5 additions & 5 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -4,10 +4,10 @@

用pytorch实现的一个简易的translator,可以根据训练的corpus来改变翻译的语言,主要包括:

1.main.py:main函数,可以选择train/test模式,自定义hidden layer、iteration、corpus、model等。<\br>
2.model.py:定义模型结构,实现了基本的RNNEncoder, RNNDecoder, AttnDecoderRNN, 详细结构见[pytorch tutorial](http://pytorch.org/tutorials/intermediate/seq2seq_translation_tutorial.html)。在原始模型上加上了中文分词(jieba)和载入word2vec pre-trained embedding(gensim)。<\br>
3.train.py:训练模型<\br>
4.test.py:用于测试模型,支持输入英文,输出对应的中文<\br>
5.load.py:用于加载语料、embedding<\br>
> * main.py:main函数,可以选择train/test模式,自定义hidden layer、iteration、corpus、model等。
> * model.py:定义模型结构,实现了基本的RNNEncoder, RNNDecoder, AttnDecoderRNN, 详细结构见[pytorch tutorial](http://pytorch.org/tutorials/intermediate/seq2seq_translation_tutorial.html)。在原始模型上加上了中文分词(jieba)和载入word2vec pre-trained embedding(gensim)。
> * train.py:训练模型
> * test.py:用于测试模型,支持输入英文,输出对应的中文
> * load.py:用于加载语料、embedding
模仿了[pytorch-chatbot](https://github.com/ywk991112/pytorch-chatbot)[pytorch tutorial](http://pytorch.org/tutorials/intermediate/seq2seq_translation_tutorial.html)的实现

0 comments on commit 24d494a

Please sign in to comment.