code: def __init__(self, *args, **kwargs):
      super().__init__(*args, **kwargs)
      self.loss_tracker = keras.metrics.Mean(name="loss")
      self.mae_metric = keras.metrics.MeanAbsoluteError(name="mae")
      self.loss_fn = keras.losses.MeanSquaredError()
  
imports: import os
  import tensorflow
  import keras
  from keras import layers
  import numpy
label_desc: Use image node_id 1 to create a new CustomModel object from inherited base classes: keras with example: obj = CustomModel()