【PyG入門學習】四:構建本身的數據集

1.簡介 雖然Pytorch-Geometric提供了不少官方數據集,可是當須要構建本身的數據集的時候,就須要對如何使用dataset基類構造本身的數據集有所瞭解。庫中提供了兩個構建數據集的基類:torch_geometric.data.Dataset和torch_geometric.data.InMemoryDataset,其中torch_geometric.data.InMemoryDatas
相關文章
相關標籤/搜索