PyTorch Geometricのデータセットの自作するための簡単なまとめ - Qiita

PyGはグラフの機械学習に有用だが、データセットの自作が最初は難しかったので最低限必要な知識を記しておく。 InMemoryDataset CPUメモリに収まるデータセットを構成できる便利なクラス。 概観 最初にインスタンスを生成すると、processメソッドが呼び出されてデータセットが処理される。 torch.save()でデータがdata.ptに保存され、次回以降高速に読み込める。 最低限知っておけばよいメソッド raw_file_names 処理前の生のデータのファイルがあればここに書いておく。 processed_file_names 処理されたデータが、__init__で渡したディレクトリの下にこのファイル名で保存される。 process グラフのノード, エッジ, 重み, 特徴量の設定などを行う。基本的にtorch_geometric.data.Dataクラスでグラフを表現し、

[sh19910711]

"PyG: グラフの機械学習に有用 + データセットの自作が最初は難しかった / InMemoryDataset: CPUメモリに収まるデータセットを構成 / インスタンスを生成すると、processメソッドが呼び出されてデータセットが処理" 2023