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 |
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 |
GenerateTabGANConfig
Configuration class for the GenerateTabGAN
action.
Attributes:
Name | Type | Description |
---|---|---|
tabular_gan |
GenerateConfig
|
Tabular GAN configuration. |