code: def __init__(self, dim, num_nodes, num_brands, num_shops, num_cates):
      super(EGES, self).__init__()
      self.dim = dim
      # embeddings for nodes
      base_embeds = th.nn.Embedding(num_nodes, dim)
      brand_embeds = th.nn.Embedding(num_brands, dim)
      shop_embeds = th.nn.Embedding(num_shops, dim)
      cate_embeds = th.nn.Embedding(num_cates, dim)
      self.embeds = [
          base_embeds,
          brand_embeds,
          shop_embeds,
          cate_embeds,
      ]
      # weights for each node's side information
      self.side_info_weights = th.nn.Embedding(num_nodes, 4)
  
imports: import torch
label_desc: Use image node_id 1 to create a new EGES object from inherited base classes: th.nn with example: obj = EGES(dim, num_nodes, num_brands, num_shops, num_cates)