code: def __init__(self, in_feats, n_hidden, out_feats, dropout):
      super(MLPX, self).__init__()
  
      self.dropout = nn.Dropout(dropout)
  
      self.fc1 = torch.nn.Linear(in_feats, n_hidden)
      self.fc2 = torch.nn.Linear(n_hidden, out_feats)
  
file_path: DIG/dig/fairgraph/method/Graphair/aug_module.py
imports: import torch
  import torch.nn
  import torch.nn.functional
  import pyro
  from .GCN import GCN_Body
label_desc: Use image node_id 1 to create a new MLPX object from inherited base classes: torch.nn with example: obj = MLPX(in_feats, n_hidden, out_feats, dropout)