Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add verbosity to TVAESynthesizer #1990

Closed
npatki opened this issue May 6, 2024 · 0 comments · Fixed by #2014
Closed

Add verbosity to TVAESynthesizer #1990

npatki opened this issue May 6, 2024 · 0 comments · Fixed by #2014
Assignees
Labels
feature request Request for a new feature
Milestone

Comments

@npatki
Copy link
Contributor

npatki commented May 6, 2024

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 npatki added the feature request Request for a new feature label May 6, 2024
@amontanez24 amontanez24 added this to the 1.13.2 milestone May 31, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
feature request Request for a new feature
Projects
None yet
Development

Successfully merging a pull request may close this issue.

3 participants