A project using Graph Neural Networks (GNNs) to classify nodes in the Cora citation network. Implements GCN and GraphSAGE models using PyTorch Geometric to classify academic papers based on citation relationships. Includes preprocessing, model training, evaluation, and visualizations.
python machine-learning deep-learning gcn citation-network cora-dataset graph-convolutional-network node-classification graphsage graph-neural-networks pytorch-geometric gnn graph-data-science research-paper-classification
-
Updated
Nov 15, 2024 - Jupyter Notebook