class_docstr: spatial feature branch
class_docstr_tok: ["spatial","feature","branch"]
code: def __init__(self, n_classes):
      super(EdgeSpatial, self).__init__()
      self.mlp = nn.Sequential()
      self.mlp.add(nn.Dense(64))
      self.mlp.add(nn.LeakyReLU(0.1))
      self.mlp.add(nn.Dense(64))
      self.mlp.add(nn.LeakyReLU(0.1))
      self.mlp.add(nn.Dense(n_classes))
  
imports: import pickle
  import dgl
  import gluoncv
  import mxnet
  import numpy
  from dgl.nn.mxnet import GraphConv
  from dgl.utils import toindex
  from mxnet import nd
  from mxnet.gluon import nn
label_desc: Use image node_id 1 to create a new EdgeSpatial object from inherited base classes: nn with example: obj = EdgeSpatial(n_classes)