src/sagemaker_sklearn_extension/contrib/taei/models.py [183:190]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
    def __init__(
        self,
        categorical_features,
        categorical_dims,
        continuous_features,
        latent_dim=32,
        hidden_dim=None,
        nll_weight=0.3,
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



src/sagemaker_sklearn_extension/contrib/taei/models.py [299:306]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
    def __init__(
        self,
        categorical_features,
        categorical_dims,
        continuous_features,
        latent_dim=32,
        hidden_dim=None,
        nll_weight=0.3,
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -