常用基准数据集 PyTorch Geometric包含了大量的常用基准数据集,例如所有的 Planetoid数据集(Cora,Citeseer,Pubmed),来自 http://graphkernels.cs.tu-dortmund.de/ 的所有经典的图数据集,QM7 和 QM9 数据集,还有少数的3D mesh/点云数据集,像 FAUST,ModelNet10/40 和 ShapeNet。初始化一个数据集非常简单直接。数...
3 changes: 2 additions & 1 deletion 3 torch_geometric/datasets/qm9.py Show comments View file Edit file Delete file This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hi...
Hello, when I use the command”from torch_geometric.loader import DataLoader“ , I will report an error”ModuleNotFoundError: No module named 'torch_geometric.loader“. What's the problem? thanks a lot. Environment PyG version:1.7.1
QM7、QM9 3D点云数据集,如FAUST、ModelNet10等 接下来拿ENZYMES数据集(包含600个图,每个图分为6个类别,图级别的分类)举例如何使用PyG的公共数据集 from torch_geometric.datasets import TUDataset # 导入数据集 dataset = TUDataset( # 指定数据集的存储位置 # 如果指定位置没有相应的数据集 # PyG会自动下载 ...