Skip to content

cornerCao/pytorch-translator

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

pytorch seq2seq translator


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

  • main.py:main函数,可以选择train/test模式,自定义hidden layer、iteration、corpus、model等。
  • model.py:定义模型结构,实现了基本的RNNEncoder, RNNDecoder, AttnDecoderRNN, 详细结构见pytorch tutorial。在原始模型上加上了中文分词(jieba)和载入word2vec pre-trained embedding(gensim)。
  • train.py:训练模型
  • test.py:用于测试模型,支持输入英文,输出对应的中文
  • load.py:用于加载语料、embedding

模仿了pytorch-chatbotpytorch tutorial的实现

About

simple translator implemented by pytorch

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages