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.

Sign up for GitHub

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 verbose parameter #2014

Merged
merged 1 commit into from
May 17, 2024
Merged

Add verbose parameter #2014

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
7 changes: 6 additions & 1 deletion sdv/single_table/ctgan.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -322,6 +322,8 @@ class TVAESynthesizer(LossValuesMixin, BaseSingleTableSynthesizer):
Regularization term. Defaults to 1e-5.
batch_size (int):
Number of data samples to process in each step.
verbose (boolean):
Whether to have print statements for progress results. Defaults to ``False``.
epochs (int):
Number of training epochs. Defaults to 300.
loss_factor (int):
Expand All @@ -338,7 +340,8 @@ class TVAESynthesizer(LossValuesMixin, BaseSingleTableSynthesizer):

def __init__(self, metadata, enforce_min_max_values=True, enforce_rounding=True,
embedding_dim=128, compress_dims=(128, 128), decompress_dims=(128, 128),
l2scale=1e-5, batch_size=500, epochs=300, loss_factor=2, cuda=True):
l2scale=1e-5, batch_size=500, verbose=False, epochs=300, loss_factor=2,
cuda=True):

super().__init__(
metadata=metadata,
Expand All @@ -350,6 +353,7 @@ def __init__(self, metadata, enforce_min_max_values=True, enforce_rounding=True,
self.decompress_dims = decompress_dims
self.l2scale = l2scale
self.batch_size = batch_size
self.verbose = verbose
self.epochs = epochs
self.loss_factor = loss_factor
self.cuda = cuda
Expand All @@ -360,6 +364,7 @@ def __init__(self, metadata, enforce_min_max_values=True, enforce_rounding=True,
'decompress_dims': decompress_dims,
'l2scale': l2scale,
'batch_size': batch_size,
'verbose': verbose,
'epochs': epochs,
'loss_factor': loss_factor,
'cuda': cuda
Expand Down
6 changes: 6 additions & 0 deletions tests/unit/single_table/test_ctgan.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -359,6 +359,7 @@ def test___init__(self):
assert instance.decompress_dims == (128, 128)
assert instance.l2scale == 1e-5
assert instance.batch_size == 500
assert instance.verbose is False
assert instance.epochs == 300
assert instance.loss_factor == 2
assert instance.cuda is True
Expand All @@ -374,6 +375,7 @@ def test___init__custom(self):
decompress_dims = (64, 64)
l2scale = 2e-5
batch_size = 250
verbose = True
epochs = 150
loss_factor = 4
cuda = False
Expand All @@ -388,6 +390,7 @@ def test___init__custom(self):
decompress_dims=decompress_dims,
l2scale=l2scale,
batch_size=batch_size,
verbose=verbose,
epochs=epochs,
loss_factor=loss_factor,
cuda=cuda,
Expand All @@ -401,6 +404,7 @@ def test___init__custom(self):
assert instance.decompress_dims == (64, 64)
assert instance.l2scale == 2e-5
assert instance.batch_size == 250
assert instance.verbose is True
assert instance.epochs == 150
assert instance.loss_factor == 4
assert instance.cuda is False
Expand All @@ -423,6 +427,7 @@ def test_get_parameters(self):
'decompress_dims': (128, 128),
'l2scale': 1e-5,
'batch_size': 500,
'verbose': False,
'epochs': 300,
'loss_factor': 2,
'cuda': True
Expand Down Expand Up @@ -459,6 +464,7 @@ def test__fit(self, mock_category_validate, mock_detect_discrete_columns, mock_t
cuda=True,
decompress_dims=(128, 128),
embedding_dim=128,
verbose=False,
epochs=300,
l2scale=1e-5,
loss_factor=2,
Expand Down
Loading