Skip to content

yibai-shi/DWGF

Repository files navigation

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:

About

A Diffusion Weighted Graph Framework for New Intent Discovery

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published