/nlp2020-project

Primary LanguagePython

Fantasy Creature Generator

Andrew Cuccinello

Submission for Natural Language Processing I

Generates descriptions of creatures with a fine-tuned GPT-2 model like the ones seen in the Pathfinder RPG or Dungeons & Dragons.

The project is organized into 4 modules, which are run in order. You can use --help to get argument information for each module.

  1. Download Data

python -m download

  1. Process Data

python -m process

  1. Train the Model

python -m train

  1. Perform generation

python -m generate