Skip to content

Latest commit

 

History

History
34 lines (26 loc) · 1 KB

README.md

File metadata and controls

34 lines (26 loc) · 1 KB

DWGF

Code for the paper titled A Diffusion Weighted Graph Framework for New Intent Discovery

Contents

1. Data

2. Requirements

3. Running

4. Thanks

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: