/PoemsFromAI

Generate poems using seq2seq model.

Primary LanguagePythonApache License 2.0Apache-2.0

PoemsFromAI

Generate poems using seq2seq model.

NOTE: We cannot guarantee any effectiveness of this model. Since nobody will maintain this repo, it will be archived from now.

Configuration

  • tensorflow 1.14 (cpu or gpu)
  • gensim

Quick Start

  1. Clone this repo.
  2. Install dependencies.
  3. Set model.GPU if you'd like to use CudnnLSTM; Else LSTMBlockFusedCell.
  4. Run word2vec.py at the first time.
  5. Run train.py. wujue is better for a test.
  6. Run compose.py. Choose only the trained model.

GUI Version

another branch

Credits

@jinfagang tensorflow_poems

License

Apache License