pyg-team
★ 24 082pytorch_geometric
Bibliothèque de réseaux neuronaux graphiques pour PyTorch. Pour cela, nous chargeons l'ensemble de données Cora et créons un modèle GCN simple à 2 couches à l'aide du GCNonv prédéfini : nous pouvons désormais optimiser le modèle dans une boucle de formation, similaire à la procédure de formation standard de PyTorch.