Graph-Neural-Networks-for-Text-Classification

example of a knowledge graph by spacy:

Training setup

15 epochs, 3e-3 lr, 128 batch size

Experiments

• Model with plain graph convolutionals train/test cross-entropy (best accuracy - 0.520):

• Model with GAT train/test cross-entropy (best accuracy - 0.525):

• Model with GATv2 from the https://arxiv.org/abs/2105.14491 train/test cross-entropy (best accuracy - 0.468):