SDV icon indicating copy to clipboard operation
SDV copied to clipboard

Add verbosity to `TVAESynthesizer`

Open npatki opened this issue 1 year ago • 0 comments

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)

npatki avatar May 06 '24 16:05 npatki