pyg-team

pyg-team

pytorch_geometric

★ 23,960

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