code: def __init__(
      self,
      guest_in_features=4,
      host_in_features=4,
      out_features=4,
      layer_lr=0.01,
      precision_bits=None,
      **kwargs,
  ):
      super().__init__()
      self._model = None
      self._layer_lr = layer_lr
      self._guest_in_features = guest_in_features
      self._host_in_features = host_in_features
      self._out_features = out_features
      self._optimizer = None
      self._precision_bits = precision_bits
  
imports: from typing import Union, Literal, Tuple
  import torch
  from fate.arch import Context
  from fate.ml.nn.model_zoo.agg_layer.agg_layer import _AggLayerBase
  from fate.arch.protocol.mpc.nn.sshe.nn_layer import SSHENeuralNetworkAggregatorLayer, SSHENeuralNetworkOptimizerSGD
label_desc: Use image node_id 1 to create a new SSHEAggLayerHost object from inherited base classes: _AggLayerBase with example: obj = SSHEAggLayerHost(guest_in_features, host_in_features, out_features, layer_lr, precision_bits)