AI-Poet

Try to train a neural network to automatically write Chinese poems

Usage

Training

You can select the model (CharRNN / Seq2Seq) by changing config.py

To train the model, just input the following command in the terminal

python main.py

Playing(×

Though so-called playing, it's not fun at all!

But you can still try it by typing the following command

python write_poem.py

References

1. LSTM_poem
2. CharRNN
3. Sequence to Sequence Learning with Neural Networks
4. Learning Phrase Representations using RNN Encoder-Decoder for Statistical Machine Translation
5. IBM/pytorch-seq2seq