使用图注意力网络的节点分类

如何解决使用图注意力网络的节点分类

我使用 GAT 模型对数据集高度不平衡的图节点进行分类。在这 80% 的数据中，我有 4 个类属于单个类。所以我无法正确训练模型。我试图给损失函数提供 class_weight 但它没有给出任何好的结果。任何人都可以提出更好的方法。我不知道我们如何利用图形的过采样概念。