teras.preprocessing module

teras.preprocessing module#

Data Transformers#

CTGANDataTransformer([continuous_features, ...])

Data Transformation class based on the data transformation in the official CTGAN paper and implementation.

GAINDataTransformer([categorical_features, ...])

GAINDataTransformer class that performs the required transformations on the raw dataset required by the GAIN architecture.

TVAEDataTransformer([continuous_features, ...])

TVAEDataTransformer class that is exactly similar to the CTGANDataTransformer, it just acts as a wrapper for convenience.

Data Samplers#

CTGANDataSampler(metadata, categorical_features)

CTGANDataSampler class based on the data sampler class in the official CTGAN implementation.

TVAEDataSampler(metadata[, ...])

TVAEDataSampler class for TVAE architecture.