synthcity.plugins.core.models.factory module

get_feature_encoder(encoder: Union[str, type], params: dict = {}) → Any

Get a feature encoder from a name or a class.

Named encoders: - datetime - onehot - label - standard - minmax - robust - quantile - bayesian_gmm - passthrough

get_model(block: Union[str, type], params: dict) → Any

Get a model from a name or a class.

Named models: - mlp - rnn - gru - lstm - transformer - tabnet

get_nonlin(nonlin: Union[str, torch.nn.modules.module.Module], params: dict = {}) → Any

Get a nonlinearity layer from a name or a class.