pyg-team
★ 23.960pytorch_geometric
Graph Neural Network Library für PyTorch. Dazu laden wir den Cora-Datensatz und erstellen ein einfaches zweischichtiges GCN-Modell mit dem vordefinierten GCNConv: Wir können das Modell nun in einer Trainingsschleife optimieren, ähnlich dem Standard-PyTorch-Trainingsverfahren.