def METHOD_NAME(self):
    config = {
        "num_classes": self.num_classes,
        "weightage": self.weightage,
        "epsilon": self.epsilon,
    }
    base_config = super().METHOD_NAME()
    return {**base_config, **config}