code: def load_tf_aegmm(
      filepath: Union[str, os.PathLike], state_dict: Dict
  ) -> tf.keras.Model:
      """
      Load AEGMM.
  
      Parameters
      ----------
      filepath
          Saved model directory.
      state_dict
          Dictionary containing the `n_gmm` and `recon_features` parameters.
  
      Returns
      -------
      Loaded AEGMM.
      """
      model_dir = Path(filepath).joinpath("model")
  
      if not [f.name for f in model_dir.glob("[!.]*.h5")]:
          logger.warning(
              "No encoder, decoder, gmm density net or aegmm found in {}.".format(
                  model_dir
              )
          )
          return None
      encoder_net = tf.keras.models.load_model(
          model_dir.joinpath("encoder_net.h5")
      )
      decoder_net = tf.keras.models.load_model(
          model_dir.joinpath("decoder_net.h5")
      )
      gmm_density_net = tf.keras.models.load_model(
          model_dir.joinpath("gmm_density_net.h5")
      )
      aegmm = AEGMM(
          encoder_net,
          decoder_net,
          gmm_density_net,
          state_dict["n_gmm"],
          state_dict["recon_features"],
      )
      aegmm.load_weights(model_dir.joinpath("aegmm.ckpt"))
      return aegmm
  
imports: import logging
  import os
  from importlib import import_module
  import warnings
  from functools import partial
  from pathlib import Path
  from typing import Callable, Dict, List, Optional, Tuple, Union, Type
  import dill
  import tensorflow
  from tensorflow_probability.python.distributions.distribution import Distribution
  from transformers import AutoTokenizer
  from alibi_detect.ad import AdversarialAE, ModelDistillation
  from alibi_detect.ad.adversarialae import DenseHidden
  from alibi_detect.cd import ChiSquareDrift, ClassifierDrift, KSDrift, MMDDrift, TabularDrift
  from alibi_detect.cd.tensorflow import UAE, HiddenOutput
  from alibi_detect.cd.tensorflow.preprocess import _Encoder
  from alibi_detect.models.tensorflow import PixelCNN, TransformerEmbedding
  from alibi_detect.models.tensorflow.autoencoder import AE, AEGMM, VAE, VAEGMM, DecoderLSTM, EncoderLSTM, Seq2Seq
  from alibi_detect.od import LLR, IForest, Mahalanobis, OutlierAE, OutlierAEGMM, OutlierProphet, OutlierSeq2Seq, OutlierVAE, OutlierVAEGMM, SpectralResidual
  from alibi_detect.od.llr import build_model
  from alibi_detect.utils.tensorflow.kernels import DeepKernel
  from alibi_detect.utils.frameworks import Framework
  from alibi_detect.version import __version__
  from alibi_detect.base import Detector
  from alibi_detect.saving._typing import VALID_DETECTORS
label_desc: Use image node_id 10 for calling a global function with example usage: load_tf_aegmm(filepath, state_dict) and returns: aegmm, None