【PyG入门学习】四:构建本身的数据集

1.简介 虽然Pytorch-Geometric提供了不少官方数据集,可是当须要构建本身的数据集的时候,就须要对如何使用dataset基类构造本身的数据集有所了解。库中提供了两个构建数据集的基类:torch_geometric.data.Dataset和torch_geometric.data.InMemoryDataset,其中torch_geometric.data.InMemoryDatas
相关文章
相关标签/搜索