From the course: Learning Graph Neural Networks

Unlock this course with a free trial

Join today to access over 24,400 courses taught by industry experts.

Exercise: The CiteSeer dataset for node classification

Exercise: The CiteSeer dataset for node classification

From the course: Learning Graph Neural Networks

Exercise: The CiteSeer dataset for node classification

- [Instructor] In this demo, we'll work with citation data represented as a graph and we'll use it in a node classification task. The nodes of the graph will represent research papers and every research paper is categorized into a class or category. So this citation dataset that we are going to be working with is similar to the Cora dataset that we'd explored earlier. In our node classification problem, we'll first perform classification using a dense neural network that only takes into account the node features. We'll use that as our baseline model. We'll then perform classification using a graph convolutional network and see how the classification model is much improved when we use GNS instead of regular DNS. Now here I'm in a new notebook, NodeClassificationWithCiteseerData. I'm going to change the runtime type so that I'm no longer using the CPU. Instead, I use the T4 GPU to run my code. Now you can run using the CPU if you want to. The dataset is small enough that it does not…

Contents