Skip to content

Training BERT transformers on synthetic data to detect Plot Holes in Fantasy Novels

License

Notifications You must be signed in to change notification settings

derekn4/plot_hole_detection

 
 

Repository files navigation

Plot Hole Detection in Fantasy Novels

This code accompanies the paper Low-Shot Learning for Fictional Claim Verification.

Steps to run:

  1. install conda env using conda_env.yml: conda env create --file=conda_env_gpu.yml
  2. install spacy lexicon: python -m spacy download en_core_web_sm
  3. see model training options and train a model: python train.py --help
  4. python clean_data.py to delete all cached data ONLY for debugging purposes (warning: this will destroy your cached data!)

Reproducibility: Despite our efforts to the contrary, reproducibility is not assured. Results that are slightly different than our own can occur with even small changes such as different CUDA versions or different GPUs (https://pytorch.org/docs/stable/notes/randomness.html for more info).

About

Training BERT transformers on synthetic data to detect Plot Holes in Fantasy Novels

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors

Languages

  • Python 99.9%
  • Shell 0.1%