Pytorch Geometric 3 - 自定义数据集
原文地址:INTRODUCTION BY EXAMPLE博客地址:陈小默的CSDNPyTorch Geometric 为数据集提供了两个抽象类torch_geometric.data.Dataset和torch_geometric.data.InMemoryDataset。其中InMemoryDataset继承自Dataset,如果要使用InMemoryDataset则需要使数据集大小适合存放在内存中。首先需要一个保存有数据文件的文件夹root,该文件夹将被划分为两个文件夹,一个用于存储数据集的.







