bidexbido / narrative-prediction

Neural network models that predict "what happens next" in stories

narrative-prediction

This repository contains Python code that performs "narrative continuation" (i.e. predicting what happens next in a story) in a few different frameworks. See each folder for a detailed description of each framework with instructions for running the code:

lm-generation:

Code that generates new sentences in stories

ROC:

Code that performs the Story Cloze Test, which predicts endings for stories

COPA:

Code that performs the Choice of Plausible Alternatives (COPA), which predicts sentences that describe a cause-effect relation between story events