GCN Definition

gcn_def.py

from dgl.nn.tensorflow import GraphConv

class GCN(tf.keras.Model):
    def __init__(self, feat_dim, hidden_dim, class_num):
        super(GCN, self).__init__()
        self.h1 = GraphConv(feat_dim, hidden_dim, activation=tf.nn.relu, allow_zero_in_degree=True)
        self.dropout = tf.keras.layers.Dropout(0.5)
        self.h2 = GraphConv(hidden_dim, class_num, allow_zero_in_degree=True)
    
    def call(self, g, features):
        h = features
        h = self.h1(g, h)
        h = self.dropout(h)
        h = self.h2(g, h)
        return h