src/sagemaker_sklearn_extension/contrib/taei/models.py [191:207]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
    ):
        super().__init__()
        if not hidden_dim:
            hidden_dim = [128, 128, 128]
        elif not isinstance(hidden_dim, list):
            hidden_dim = [
                hidden_dim,
            ]

        assert len(categorical_features) == len(categorical_dims)
        self.categorical_features, self.categorical_dims = [], []
        if categorical_features and categorical_dims:
            self.categorical_features, self.categorical_dims = zip(*sorted(zip(categorical_features, categorical_dims)))
        self.continuous_features = sorted(continuous_features)
        self.latent_dim = latent_dim
        self.hidden_dim = hidden_dim
        self.nll_weight = nll_weight
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



src/sagemaker_sklearn_extension/contrib/taei/models.py [308:324]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
    ):
        super().__init__()
        if not hidden_dim:
            hidden_dim = [128, 128, 128]
        elif not isinstance(hidden_dim, list):
            hidden_dim = [
                hidden_dim,
            ]

        assert len(categorical_features) == len(categorical_dims)
        self.categorical_features, self.categorical_dims = [], []
        if categorical_features and categorical_dims:
            self.categorical_features, self.categorical_dims = zip(*sorted(zip(categorical_features, categorical_dims)))
        self.continuous_features = sorted(continuous_features)
        self.latent_dim = latent_dim
        self.hidden_dim = hidden_dim
        self.nll_weight = nll_weight
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -