Skip to content

Latest commit

 

History

History
8 lines (6 loc) · 283 Bytes

README.md

File metadata and controls

8 lines (6 loc) · 283 Bytes

DialogueHGT

Using heterogeneous graph transformer to replace RGCN in DialogueGCN.

To reproduce the IEMOCAP dataset results, run:

python train_iemocap_hgt.py --base-model 'LSTM' --nodal-attention --dropout 0.4 --lr 0.0003 --batch-size 32 --class-weight --l2 0.0 --no-cuda