Human-machine interactive storytelling engine finetuned for scary/creepy/chilling stories. Finetuned on GPT2, GPT-NEO, and a custom pretrained GPT2 transformer networks on custom corpus for the downstream task of text generation. Integrated final model into a chatbot front end for human-machine interface.
See App Execution for details
Stories were sourced from the following sub-reddits
Number | Subreddit | Link |
---|---|---|
1. | r/nosleep | Link |
2. | r/stayawake | Link |
3. | r/DarkTales | Link |
4. | r/LetsNotMeet | Link |
5. | r/shortscarystories | Link |
6. | r/Thetruthishere | Link |
7. | r/creepyencounters | Link |
8. | r/truescarystories | Link |
9. | r/Glitch_in_the_Matrix | Link |
10. | r/Paranormal | Link |
11. | r/Ghoststories | Link |
12. | r/libraryofshadows | Link |
13. | r/UnresolvedMysteries | Link |
14. | r/TheChills | Link |
15. | r/scaredshitless | Link |
16. | r/scaryshortstories | Link |
17. | r/Humanoidencounters | Link |
18. | r/DispatchingStories | Link |
See Data Acquisition section for more details.
- Code: This directory holds all relevant code for data acquisition, preprocessing, model building/training, evaluation, and front end.
- Group-Proposal: Group proposal description for project.
- .gitignore: gitignore file
- LICENSE: license description
- README.md: readme file
- requirements.txt: python requirements
- Conditional Text Generation for Harmonious Human-Machine Interaction, Guo et al., 2020
- Transformers: State-of-the-Art Natural Language Processing, Wolf et al., 2020
@inproceedings{wolf-etal-2020-transformers,
title = "Transformers: State-of-the-Art Natural Language Processing",
author = "Thomas Wolf and Lysandre Debut and Victor Sanh and Julien Chaumond and Clement Delangue and Anthony Moi and Pierric Cistac and Tim Rault and Rémi Louf and Morgan Funtowicz and Joe Davison and Sam Shleifer and Patrick von Platen and Clara Ma and Yacine Jernite and Julien Plu and Canwen Xu and Teven Le Scao and Sylvain Gugger and Mariama Drame and Quentin Lhoest and Alexander M. Rush",
booktitle = "Proceedings of the 2020 Conference on Empirical Methods in Natural Language Processing: System Demonstrations",
month = oct,
year = "2020",
address = "Online",
publisher = "Association for Computational Linguistics",
url = "https://www.aclweb.org/anthology/2020.emnlp-demos.6",
pages = "38--45"
}
- Transformer-based Conditional Variational Autoencoder for Controllable Story Generation, Fang et al., 2021
- https://arrow.tudublin.ie/cgi/viewcontent.cgi?article=1214&context=scschcomdis, Araz, 2020
- Improving Neural Story Generation by Targeted Common Sense Grounding, hhmao et al., 2019
- Evaluation of Text Generation: A Survey, Celikyilmaz et al., 2021