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

时间:2021-04-07 17:00:24

标签: python graph artificial-intelligence classification

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

0 个答案:

没有答案
相关问题