Event Entity Coreference with SpanBERT on ECB+.
More details will be added later.
More details about the ECB+ dataset can be found here.
To download the dataset, click here.
Python 3.8.5
Pytorch
Transformers
Exact versions and other details are mentioned in requirements.txt
main.py - run this file with required options
src / model.py - has the pytorch model class
src / train.py - has methods for training, evaluation (will rename this later)
src / dataloader.py - has pytorch dataset class
src / utils.py - has all the misc functions used for data processing and other stuff.
python main.py -h for usage details
More details regarding the repository will be added soon.