code: def __init__(
      self,
      base_model: tf.keras.Model,
      units: int,
      chain_initializer: types.Initializer = "orthogonal",
      use_boundary: bool = True,
      boundary_initializer: types.Initializer = "zeros",
      use_kernel: bool = True,
      **kwargs,
  ):
      super().__init__()
  
      # lazy import to solve circle import issue:
      # tfa.layers.CRF -> tfa.text.__init__ -> tfa.text.crf_wrapper -> tfa.layers.CRF
      from tensorflow_addons.layers.crf import CRF  # noqa
  
      self.crf_layer = CRF(
          units=units,
          chain_initializer=chain_initializer,
          use_boundary=use_boundary,
          boundary_initializer=boundary_initializer,
          use_kernel=use_kernel,
          **kwargs,
      )
  
      self.base_model = base_model
  
file_path: addons/tensorflow_addons/text/crf_wrapper.py
imports: import tensorflow
  from tensorflow_addons.text import crf_log_likelihood
  from tensorflow_addons.utils import types
label_desc: Use image node_id 1 to create a new CRFModelWrapper object from inherited base classes: tf.keras with example: obj = CRFModelWrapper(base_model, units, chain_initializer, use_boundary, boundary_initializer, use_kernel)