code: def __init__(self, input_size: int, num_class: int):
      super(MulticlassNetwork, self).__init__()
  
      self.linear_relu_stack = nn.Sequential(
          nn.Linear(input_size, 16), nn.ReLU(), nn.Linear(16, num_class)
      )
      self.softmax = nn.Softmax(dim=1)
  
file_path: DiCE/dice_ml/utils/neuralnetworks.py
imports: from torch import nn, sigmoid
label_desc: Use image node_id 1 to create a new MulticlassNetwork object from inherited base classes: nn with example: obj = MulticlassNetwork(input_size, num_class)