def METHOD_NAME(self) -> Dict[str, Any]:
    """
    Returns a dictionary representation of the model configuration.
    """
    return {
        "model": self.name(),
        **{
            (
                k
                if isinstance(self, TrainConfig)
                and k in TrainConfig.training_hyperparameters()
                else f"{self.name()}_{k}"
            ): v
            for k, v in METHOD_NAME(self).items()
        },
    }