Skip to content

Generative LLM specifically trained to generate poems This LLM uses Pytorch's GPT2 and a dataset of poems from The Poetry Foundation to train a neural network that generates new poetry based on a prompt.

Notifications You must be signed in to change notification settings

zhoucaiNi/poet-gpt-2

Repository files navigation

Poet-GPT

Files

  • modelLoading.ipynb - responsible for loading the trained model
  • modelTraining.ipynb - responsible for training the language model
  • PoetryFoundationData.csv - the main training database
  • ModelReport.pdf - contains additional details and results for the project

To download the best model I trained click this link

Some example outputs:

Screen Shot 2023-06-05 at 11 05 00 PM Screen Shot 2023-06-05 at 11 04 48 PM

About

Generative LLM specifically trained to generate poems This LLM uses Pytorch's GPT2 and a dataset of poems from The Poetry Foundation to train a neural network that generates new poetry based on a prompt.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published