Skip to content

rockfish.actions.tab_gan

rockfish.actions.tab_gan

Classes

TrainTabGANConfig

Configuration class for the TrainTabGAN action.

Attributes:

Name Type Description
tabular_gan TrainConfig

Tabular GAN configuration.

encoder DatasetConfig

Dataset configuration.

model_labels LabelDict

Additional labels to add to the Model.

TrainConfig

Training configuration.

Attributes:

Name Type Description
batch_size int

Number of data samples to process in each step. This value must be even, and it must be divisible by the pac parameter.

epochs int

Number of training epochs.

pac int

Number of samples to group together when applying the discriminator.

min_max Optional[bool]

This parameter is no longer in effect. Please set clip_in_range in the generation config instead.

GenerateTabGANConfig

Configuration class for the GenerateTabGAN action.

Attributes:

Name Type Description
tabular_gan GenerateConfig

Tabular GAN configuration.

GenerateConfig

Generate configuration.

Attributes:

Name Type Description
records Optional[int]

Number of records to generate.

clip_in_range bool

Whether to clip generated continuous values to be within the range of the training data.