code: def __init__(
      self,
      in_features,
      n_binary=0,
      n_categorical=0,
      n_numerical=0,
      cardinality=[],
  ):
      super().__init__()
      assert n_categorical == len(
          cardinality
      ), "require cardinalities for each categorical variable"
      assert n_binary + n_categorical + n_numerical, "need some targets"
      self.n_binary = n_binary
      self.n_categorical = n_categorical
      self.n_numerical = n_numerical
  
      self.binary_linear = (
          nn.Linear(in_features, n_binary) if n_binary else None
      )
      self.categorical_linears = nn.ModuleList(
          [nn.Linear(in_features, card) for card in cardinality]
      )
      self.numerical_linear = (
          nn.Linear(in_features, n_numerical) if n_numerical else None
      )
  
imports: import torch.nn
label_desc: Use image node_id 1 to create a new MultiTaskHead object from inherited base classes: nn with example: obj = MultiTaskHead(in_features, n_binary, n_categorical, n_numerical, cardinality)