Skip to content

SeqGAN for poem generation, can be modified to fit the lyrics genertion problem

Notifications You must be signed in to change notification settings

AWLyrics/SeqGAN_Poem

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

10 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

SeqGAN Poem

修改 SeqGAN 用于诗歌生成(去除 Oracle LSTM)

Pipeline

  • 收集唐诗预处理(preprocessing.py),唐诗数据集,选取了 5000 首五言律诗(20 词)
  • Tokenize,将中文字转换为 index(train.txt),建立词表(dict.pkl),记录 vocab size
  • Pretrain Generation
  • Adersarial Training

Reference

Paper: https://arxiv.org/abs/1609.05473

Original Repo: SeqGAN

Blog: SeqGAN——GAN + RL + NLP

About

SeqGAN for poem generation, can be modified to fit the lyrics genertion problem

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published