Global and Local Hierarchy-aware Contrastive Framework for Hierarchical Implicit Discourse Relation Recognition
First, install PyTorch by following the instructions from the official website. To faithfully reproduce our results, please use the correct 1.10.1
version corresponding to your platforms/CUDA versions. PyTorch version higher than 1.10.1
should also work.
Then run the following script to install the remaining dependencies,
pip install -r requirements.txt
- Download the PDTB2.0 dataset, put it under /raw/
- Run the following script for data preprocessing,
python3 preprocess.py
Run the following script for training, evaludating, and testing,
python3 run.py
Please cite our paper by:
@misc{jiang2022promcse,
title={Improved Universal Sentence Embeddings with Prompt-based Contrastive Learning and Energy-based Learning},
author={Yuxin Jiang, Linhan Zhang and Wei Wang},
year={2022},
eprint={2203.06875},
archivePrefix={arXiv},
primaryClass={cs.CL}
}