首页 > 解决方案 > 使用图注意力网络进行节点分类

问题描述

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

标签: pythongraphartificial-intelligenceclassification

解决方案


推荐阅读