pytorch_geometric

PyTorch 用のグラフ ニューラル ネットワーク ライブラリ。このために、Cora データセットをロードし、事前定義された GCNConv を使用して単純な 2 層 GCN モデルを作成します。標準の PyTorch トレーニング手順と同様に、トレーニング ループでモデルを最適化できるようになりました。

このページのプロジェクト情報と編集内容は無料で読めます。元の GitHub リポジトリが最終的な情報源であり、保存や議論への参加時だけログインが必要です。