Skip to content

knok/chainer-fm-gan

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

32 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

chainer-fm-gan

This is a re-implimentation of FM-GAN using chainer.

Requirements:

  • Python 3.5+
  • Chainer 5.2+
  • wget, nkf, mecab (for preprocessing dataset)

How to run

Dataset

data/ directory contains scripts to download the Livedoor news corpus and preprocess them.

Just type the following:

cd data && make all

Train

export GPU=0 # if you want to use GPU
./train.sh # pretrain LSTM-VAE
./gantrain.sh # GAN training

License

3-clause BSD

About

Re-implementation of FM-GAN, based on the TensorFlow code

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published