Creating Custom Graph - Confused with RGCN Tutorial

christina-s-wang · December 15, 2020, 10:25pm

Hello DGL Community!

I am new to DGL and am following the RGCN tutorial here: https://docs.dgl.ai/en/0.4.x/tutorials/models/1_gnn/4_rgcn.html

The tutorial walks through the AIFB dataset, and I tried modifying the tutorial notebook to work with my own dataset. However, I am confused as to what the parameters for the model mean and am having a hard time figuring out what they are (e.g. num_rels, num_classes, labels).

I assumed that num_rels was the number of unique edge types, but when I look at the AIFB unique edge types versus data.num_rels, the numbers do not match. Also, I’m guessing that num_classes is the number of node classes?

My Jupyter notebook is in the link below and is completed to the best of my guessing, which runs, but it sigkills on my end without any error messages. I made the train, validation, and test batches very small (10, 5, 5) so I’m not sure why this is happening!

github.com

christina-s-wang/rgcn_testing/blob/main/DGL_Relational_Graph_Convolutional_Network.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [],
   "source": [
    "%matplotlib inline"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "(R-GCN): This type of network is one effort to generalize GCN \n",
    "to handle different relationships between entities in a knowledge base. \n",
    "To learn more about the research behind R-GCN, see `Modeling Relational Data with Graph Convolutional\n",
    "Networks <https://arxiv.org/pdf/1703.06103.pdf>`_ \n",
    "\n",

This file has been truncated. show original

Help would be much appreciated!

mufeili · December 16, 2020, 7:41am

Sorry for the unpleasant experience.

num_rels is the number of relations. It seems that Model takes data.num_rels for the num_rels argument here?
num_classes is the number of node classes as you said.
Can you also upload ‘network.csv’?

Do you have any suggestions for improving the tutorial? E.g. will it be better to have doc for the parameters in the tutorial?

system · January 15, 2021, 7:41am

This topic was automatically closed 30 days after the last reply. New replies are no longer allowed.