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: Setting up a DNN as a baseline model

Exercise: Setting up a DNN as a baseline model

From the course: Learning Graph Neural Networks

Exercise: Setting up a DNN as a baseline model

- [Instructor] Our objective in this machine learning problem is to perform node classification. You want to classify the research papers into whether they're about agents, machine learning, AI, or any of the six classes or categories that we have. Before we use graph convolution networks, the first thing that we'll do is set up a simple dense neural network, a multi-layer perceptron to classify the nodes into categories. Now, in theory, we should be able to infer the category of a document solely based on its contents. That is its bag of words feature representation without taking any relationship between nodes into account. If you remember, we have a feature vector associated with each node representing the presence or absence of words from a predefined dictionary. In the research paper represented by that node, I'll now set up a simple dense neural network to perform node classification based only on the node features without taking the structure of the graph into account, without…

Contents