Denken
Menu
  • About Me
  • Deep Learning with Pytorch
  • Generative AI: Tutorial Series
  • Python Tutorials
  • Contact Me
Menu

Graph Neural Network – Node Classification with PyG – 2.1

Posted on November 10, 2022November 11, 2022 by Aritra Sen

After covering the basics of GNN and PyG , now let’s start doing the actual implementation of GNN model training and inference. In this post we will use Cora citation dataset, where each node is a paper and edges refer to the citation. There seven types of papers (Example: Paper of DL, RL, Graphs) which is our target variable. Each node is a representation of Bag-of-words for the words present in the individual paper. Bag-of-words or the node features are 1433-dimensional vector. Based on these features and edge connections related to citation, when a unknow type of paper comes into the network, we have to classify the type of paper. Below shown image can describe the problem in our hand where we have to predict the category of the new paper which probably should be Graph Neural Network as it cited the Deep Learning and Graph Mining papers.

Citation Knowledge Graph

First, we will explore the dataset then train a GCN network for node classification and finally we will also train a multilayer perceptron to compare the performance on the model in the test set. There are smaller number of datapoints to train, that’s why we are not batching the dataset instead directly running the epochs.

Do like, share and comment if you have any questions.

Category: Machine Learning, Python

Post navigation

← Graph Neural Network – Pytorch Geometric (PyG) Intro – 2.0
NLP: Toxic comment classification with TorchText & Pytorch Lightning →

Leave a Reply Cancel reply

Your email address will not be published. Required fields are marked *

RSS Feeds

Enter your email address:

Delivered by FeedBurner

Pages

  • About Me
  • Contact Me
  • Deep Learning with Pytorch
  • Generative AI: Tutorial Series
  • Python Tutorials

Tag Cloud

Announcements Anrdoid BERT Bias Celebration Cricket CyanogenMod deep-learning Denken Experience Facebook Features Finetuning GCN GenerativeAI GNN Google HBOOT HBOOT downgrading HTC Wildfire huggingface India Launch Life LLM Lumia 520 MachineLearning mobile My Space nlp Orkut People Python pytorch pytorch-geometric Rooting Sachin Share Social Network tranformers transformers Tutorials Twitter weight-initialization Windows Phone

WP Cumulus Flash tag cloud by Roy Tanck and Luke Morton requires Flash Player 9 or better.

Categories

Random Posts

  • Graph Neural Network – Getting Started – 1.0
  • Deep Learning with Pytorch-nn.Sequential,GPU,Saving & Loading Model – 1.3
  • How to upload pictures in Flickr/Facebook from Picasa
  • Generative AI: LLMs: LoRA fine tuning 1.4
  • Python Tutorials – 1.6 – Class and Instances

Recent Comments

  • Generative AI: LLMs: Reduce Hallucinations with Retrieval-Augmented-Generation (RAG) 1.8 – Denken on Generative AI: LLMs: Semantic Search and Conversation Retrieval QA using Vector Store and LangChain 1.7
  • vikas on Domain Fuss
  • Kajal on Deep Learning with Pytorch -Text Generation – LSTMs – 3.3
  • Aritra Sen on Python Tutorials – 1.1 – Variables and Data Types
  • Aakash on Python Tutorials – 1.1 – Variables and Data Types

Visitors Count

AmazingCounters.com

Archives

Meta

  • Log in
  • Entries feed
  • Comments feed
  • WordPress.org

Copyright

AritraSen’s site© This site has been protected from copyright by copyscape.Copying from this site is stricktly prohibited. Protected by Copyscape Original Content Validator
© 2025 Denken | Powered by Minimalist Blog WordPress Theme