SDV
SDV copied to clipboard
Add verbosity to `TVAESynthesizer`
Problem Description
As discussed in CTVAN issue 300: The TVAE model has a verbose parameter that you can set to see a progress bar per epoch.
# base model TVAE has a verbose parameter
from ctgan.synthesizers import TVAE
model = TVAE(verbose=True, epochs=10)
model.fit(processed_data, discrete_columns=('has_rewards', 'room_type'))
Loss: 1.730: 100%|██████████| 10/10 [00:19<00:00, 1.91s/it]
However, the SDV wrapper around it (TVAESynthesizer) does not expose this parameter, so you cannot access verbosity through SDV.
Expected behavior
The TVAESynthesizer (SDV wrapper) should also expose the verbose parameter and use it to set verbosity on the underlying model. This should work exactly as verbosity works for CTGANSynthesizer.
from sdv.single_table import TVAESynthesizer
synth = TVAESynthesizer(metadata, verbose=True, epochs=10)
synth.fit(data)