1.InMemoryDataset基类
在PyG中,可以通过继承InMemoryDataset类来自定义一个数据可全部存储到内存的数据集类。(继承Dataset是分次加载到内存,继承InMemoryDataset是一次性加载所有数据到内存)
1 | class InMemoryDataset(root: Optional[str] = None, transform: Optional[Callable] = None, pre_transform: Optional[Callable] = None, pre_filter: Optional[Callable] = None) |
参数说明:
transform:数据转换函数,用于转换Data对象,每一次数据获取过程中都会被执行。
pre_transform:数据转换函数,用于转换Data对象,在Data对象被保存到文件前调用。
pre_filter:检查数据是否要保留的函数,接收一个Data对象,返回此Data对象是否应该被包含在最终的数据集中,在Data对象被保存到文件前调用。