teras.models.CTGANDiscriminator#
- class teras.models.CTGANDiscriminator(hidden_dims=(256, 256), packing_degree=8, gradient_penalty_lambda=10.0, seed=1337, **kwargs)[source]#
CTGANDiscriminator for CTGAN architecture as proposed by Lei Xu et al. in the paper, “Modeling Tabular data using Conditional GAN”.
- Reference(s):
- Parameters:
hidden_dims (
Union[List[int],Tuple[int]]) – Sequence, A sequence of integers that is used to construct the hidden block. For each value, a CTGANDiscriminatorLayer of that dimensionality is added. Defaults to [256, 256]packing_degree (
int) – int, Packing degree - taken from the PacGAN paper. The number of samples concatenated or “packed” together. It must be a factor of the batch_size. Packing degree is borrowed from the PacGAN [Diederik P. Kingma et al.] architecture, which proposes passing m samples at once to discriminator instead of 1 to be jointly classified as real or fake by the discriminator in order to tackle the issue of mode collapse inherent in the GAN based architectures. The number of samples passed jointly m, is termed as the packing degree. Defaults to 8.gradient_penalty_lambda (
float) – float, Controls the strength of gradient penalty. lambda value is directly proportional to the strength of gradient penalty. Gradient penalty penalizes the discriminator for large weights in an attempt to combat Discriminator becoming too confident and overfitting. Defaults to 10.seed (
int) – int, Seed for random sampling. Defaults to 1337.
- __init__(hidden_dims=(256, 256), packing_degree=8, gradient_penalty_lambda=10.0, seed=1337, **kwargs)[source]#
Methods
__init__([hidden_dims, packing_degree, ...])add_loss(loss)Can be called inside of the call() method to add a scalar loss.
add_metric()add_variable(shape, initializer[, dtype, ...])Add a weight variable to the layer.
add_weight([shape, initializer, dtype, ...])Add a weight variable to the layer.
build(input_shape)build_from_config(config)Builds the layer's states with the supplied config dict.
call(inputs)compile([optimizer, loss, loss_weights, ...])Configures the model for training.
compile_from_config(config)Compiles the model with the information given in config.
compiled_loss(y, y_pred[, sample_weight, ...])compute_loss([x, y, y_pred, sample_weight, ...])Compute the total loss, validate it, and return it.
compute_mask(inputs, previous_mask)compute_metrics(x, y, y_pred[, sample_weight])Update metric states and collect all metrics to be returned.
compute_output_shape(input_shape)compute_output_spec(*args, **kwargs)count_params()Count the total number of scalars composing the weights.
evaluate([x, y, batch_size, verbose, ...])Returns the loss value & metrics values for the model in test mode.
export(filepath[, format])Create a TF SavedModel artifact for inference.
fit([x, y, batch_size, epochs, verbose, ...])Trains the model for a fixed number of epochs (dataset iterations).
from_config(config[, custom_objects])Creates an operation from its config.
get_build_config()Returns a dictionary with the layer's input shape.
get_compile_config()Returns a serialized config with information for compiling the model.
get_config()Returns the config of the object.
get_layer([name, index])Retrieves a layer based on either its name (unique) or index.
get_metrics_result()Returns the model's metrics values as a dict.
get_weights()Return the values of layer.weights as a list of NumPy arrays.
gradient_penalty(real_samples, generated_samples)load_own_variables(store)Loads the state of the layer.
load_weights(filepath[, skip_mismatch])Load weights from a file saved via save_weights().
loss(y, y_pred[, sample_weight])make_predict_function([force])make_test_function([force])make_train_function([force])predict(x[, batch_size, verbose, steps, ...])Generates output predictions for the input samples.
predict_on_batch(x)Returns predictions for a single batch of samples.
predict_step(data)quantize(mode)Quantize the weights of the model.
quantized_call(*args, **kwargs)reset_metrics()save(filepath[, overwrite, zipped])Saves a model as a .keras file.
save_own_variables(store)Saves the state of the layer.
save_weights(filepath[, overwrite])Saves all layer weights to a .weights.h5 file.
set_weights(weights)Sets the values of layer.weights from a list of NumPy arrays.
stateless_call(trainable_variables, ...[, ...])Call the layer without any side effects.
stateless_compute_loss(trainable_variables, ...)summary([line_length, positions, print_fn, ...])Prints a string summary of the network.
symbolic_call(*args, **kwargs)test_on_batch(x[, y, sample_weight, return_dict])Test the model on a single batch of samples.
test_step(data)to_json(**kwargs)Returns a JSON string containing the network configuration.
train_on_batch(x[, y, sample_weight, ...])Runs a single gradient update on a single batch of data.
train_step(data)Attributes
compiled_metricscompute_dtypeThe dtype of the computations performed by the layer.
distribute_reduction_methoddistribute_strategydtypeAlias of layer.variable_dtype.
dtype_policyinputRetrieves the input tensor(s) of a symbolic operation.
input_dtypeThe dtype layer inputs should be converted to.
input_specjit_compilelayerslossesList of scalar losses from add_loss, regularizers and sublayers.
metricsList of all metrics.
metrics_namesmetrics_variablesList of all metric variables.
non_trainable_variablesList of all non-trainable layer state.
non_trainable_weightsList of all non-trainable weight variables of the layer.
outputRetrieves the output tensor(s) of a layer.
pathThe path of the layer.
quantization_modeThe quantization mode of this layer, None if not quantized.
run_eagerlysupports_maskingWhether this layer supports computing a mask using compute_mask.
trainableSettable boolean, whether this layer should be trainable or not.
trainable_variablesList of all trainable layer state.
trainable_weightsList of all trainable weight variables of the layer.
variable_dtypeThe dtype of the state (weights) of the layer.
variablesList of all layer state, including random seeds.
weightsList of all weight variables of the layer.