Skip to content

Using heterogeneous graph transformer in DialogueGCN.

Notifications You must be signed in to change notification settings

DesmonDay/DialogueHGT

Repository files navigation

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

About

Using heterogeneous graph transformer in DialogueGCN.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages