DWGF
December 21, 2023 ยท View on GitHub
Code for the paper titled A Diffusion Weighted Graph Framework for New Intent Discovery
Contents
Data
We performed experiments on three public datasets: clinc, banking and stackoverflow, which have been included in our repository in the data folder ' ./data '.
Requirements
- python==3.8
- pytorch==1.12.0
- transformers==4.26.1
- scipy==1.10.1
- numpy==1.23.5
- scikit-learn==1.2.1
Running
Pre-training, training, and inference through the following scripts:
bash scripts/run_{dataset}.sh
Thanks
Some code references the following repositories: