neural-network - 如何在pytorch几何中定义类的数量

Question

我是图形神经网络的新手，我正在尝试使用 pytorch-geometric 来读取 sdf 文件以进行预测，

我有以下数据集作为我的输入

Data(edge_attr=[52, 6], edge_index=[2, 52], x=[24, 82])
Data(edge_attr=[60, 6], edge_index=[2, 60], x=[26, 82])
Data(edge_attr=[50, 6], edge_index=[2, 50], x=[23, 82])
Data(edge_attr=[48, 6], edge_index=[2, 48], x=[22, 82])
...

我希望我的目标预测看起来像

14
17
22
20
...

如您所见，目标小于或等于输入 x 中的节点数，其中 x = [num_nodes, num_node_features]。

我想知道如何为这样的数据集构建图神经网络？

我应该调整输入 x 的大小还是其他？

谢谢！