spektral.datasets.引用.load_data() 未找到

Question

我正在尝试使用 CoRA 数据集在张量流上训练图神经网络，这是我第一次使用 Spektral。

经过在互联网上的一些研究，我了解到 Spektral 应该有一个有用的加载器函数供我加载这个基准数据集，所以我尝试实现它：

adj,features,labels,train_mask,val_mask,test_mask=spektral.datasets.citation.load_data(dataset_name="cora")

然后我得到了这个：

AttributeError: module 'spektral.datasets.citation' has no attribute 'load_data'

我检查了 Spektral 文档，但没有找到load_data加载器功能，所以我认为有更新或其他内容。

如果任何人都可以提供任何替代方案或可能的错误解释，那将会很有帮助。

Answer 1

经过更多研究后，我自己解决了这个问题。对于 Spektral 版本 1.1.0，我认为这在使用 CoRA 数据集方面完成了加载器函数的工作：

cora_dataset = spektral.datasets.citation.Citation(name='cora')
test_mask = cora_dataset.mask_te
train_mask = cora_dataset.mask_tr
val_mask = cora_dataset.mask_va
graph = cora_dataset.graphs[0]
features = graph.x
adj = graph.a
labels = graph.y