The code in this repository is based on the brain-lang repository, but contains new developments and analyses on story representations. It is work in progress.
- Numpy
- Sklearn
- Allennlp (for Elmo)
- pytorch_pretrained_bert (for Bert encoder)
- Spacy for tokenization: python -m spacy download en_core_web_lg
- Pandas, matplotlib, seaborn and nilearn in case you want to plot the results.