code: def __init__(self, feat=18, class_num=4) -> None:
      super().__init__()
      self.class_num = class_num
      self.model = nn.Sequential(
          nn.Linear(feat, 10), nn.ReLU(), nn.Linear(10, class_num)
      )
  
imports: from torch import nn
label_desc: Use image node_id 1 to create a new Multi object from inherited base classes: nn with example: obj = Multi(feat, class_num)