Skip to content

pmlg/pmlg-poem-generator

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

7 Commits
 
 
 
 
 
 
 
 

Repository files navigation

pmlg-poem-generator

This is a poem generator trained with Gated Recurrent Units (GRU) on a corpus of poems including those in:

Most of the code comes from Tensorflow's text generation tutorial.

Please refer to the notebook for an overview of the results.

Improvement ideas

Short term:

  • Clean the corpus
  • Find more data
  • Save a model
  • Try LSTM
  • Train a model based on sequences of words instead of characters

Middle term goals:

  • Create a rhyme generator
  • Train a model backward: from the last word to the first

The idea is to generate the rhymes first, and then write the lines from the last word to the first. This process makes sure that the generated poems rhyme.

Long term goals

  • Generate other kind of text (Haiku, magazines, articles, books...)

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Jupyter Notebook 82.0%
  • Python 18.0%