Source code for pythae.models.rae_gp.rae_gp_config

from pydantic.dataclasses import dataclass

from ..ae import AEConfig


@dataclass
class RAE_GP_Config(AEConfig):
    """RAE_GP config class.

    Parameters:
        input_dim (tuple): The input_data dimension.
        latent_dim (int): The latent space dimension. Default: None.
        embedding_weight (float): The factor before the L2 embedding regularization term in the
            loss. Default: 1e-4
        reg_weight (float): The factor before the gradient penalty regularization term.
    """

    embedding_weight: float = 1e-4
    reg_weight: float = 1e-7