code: def __init__(self, in_feats, h_feats, num_classes):
      super(Model, self).__init__()
      self.conv1 = WeightedSAGEConv(in_feats, h_feats)
      self.conv2 = WeightedSAGEConv(h_feats, num_classes)
  
imports: import os
  import dgl
  import dgl.function
  import torch
  import torch.nn
  import torch.nn.functional
  import dgl.data
label_desc: Use image node_id 1 to create a new Model object from inherited base classes: nn with example: obj = Model(in_feats, h_feats, num_classes)