code: def __init__(
      self,
      num_nodes,
      h_dim,
      out_dim,
      num_rels,
      num_bases=-1,
      num_hidden_layers=1,
  ):
      super(Model, self).__init__()
      self.num_nodes = num_nodes
      self.h_dim = h_dim
      self.out_dim = out_dim
      self.num_rels = num_rels
      self.num_bases = num_bases
      self.num_hidden_layers = num_hidden_layers
  
      # create rgcn layers
      self.build_model()
  
      # create initial features
      self.features = self.create_features()
  
imports: import os
  from functools import partial
  import dgl
  import dgl.function
  import torch
  import torch.nn
  import torch.nn.functional
  from dgl import DGLGraph
label_desc: Use image node_id 1 to create a new Model object from inherited base classes: nn with example: obj = Model(num_nodes, h_dim, out_dim, num_rels, num_bases, num_hidden_layers)