Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
28 commits
Select commit Hold shift + click to select a range
21bb7e8
wip
sarahmish Jul 16, 2026
4be6278
add tabddpm for single-table synthesis
sarahmish Jul 17, 2026
f18a5ca
update synthesizer list test
sarahmish Jul 17, 2026
431e053
update code
sarahmish Jul 17, 2026
68feb7f
add clavaddpm and tests
sarahmish Jul 22, 2026
a185358
add benchmark_multi_table test
sarahmish Jul 22, 2026
f71f40a
fix lint
sarahmish Jul 22, 2026
21d0094
fix sklearn onhotencoding version
sarahmish Jul 27, 2026
6ef5fa4
change init_param based on sklearn version
sarahmish Jul 27, 2026
e3513c2
kmeans
sarahmish Jul 27, 2026
f312993
add space
sarahmish Jul 27, 2026
8ce08a5
create primary key if none exist
sarahmish Jul 30, 2026
bc076e2
add id column to discrete
sarahmish Jul 31, 2026
288b95b
address comments
sarahmish Aug 3, 2026
8224ad1
Merge branch 'main' into clavaddpm
sarahmish Aug 3, 2026
0f05bf7
change to in-house datetime guesser
sarahmish Aug 4, 2026
72ecee0
import guess_datetime_format depending on version
sarahmish Aug 5, 2026
29fee25
cap categorical columns to 100
sarahmish Aug 6, 2026
b201719
cast mixed formats to be uniform
sarahmish Aug 6, 2026
8048265
move preprocessing to fit only
sarahmish Aug 10, 2026
d1340b2
skip validation on tabddpm if run from clavaddpm
sarahmish Aug 13, 2026
33ca690
reorganize code
sarahmish Aug 18, 2026
5fa3b40
address comments
sarahmish Aug 18, 2026
8aff16c
Merge branch 'main' into clavaddpm
sarahmish Aug 18, 2026
ec384ac
fix lint
sarahmish Aug 18, 2026
64c84c7
fix index
sarahmish Aug 18, 2026
dfdae00
seperate faiss req
sarahmish Aug 18, 2026
8e47e85
fix
sarahmish Aug 18, 2026
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
5 changes: 5 additions & 0 deletions pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -85,8 +85,13 @@ realtabformer = [
"torch>=2.6.0",
'transformers<4.51',
]
clavaddpm = [
'faiss-cpu>=1.10.0',
'numpy>=1.25.0',
]
test = [
'sdgym[realtabformer]',
'sdgym[clavaddpm]',
'pytest>=6.2.5',
'pytest-cov>=2.6.0',
'jupyter>=1.0.0,<2',
Expand Down
2 changes: 2 additions & 0 deletions sdgym/synthesizers/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,6 +9,7 @@
from sdgym.synthesizers.column import ColumnSynthesizer
from sdgym.synthesizers.realtabformer import RealTabFormerSynthesizer
from sdgym.synthesizers.tabddpm import TabDDPMSynthesizer
from sdgym.synthesizers.clavaddpm import ClavaDDPMSynthesizer
from sdgym.synthesizers.uniform import UniformSynthesizer, MultiTableUniformSynthesizer
from sdgym.synthesizers.utils import (
get_available_single_table_synthesizers,
Expand All @@ -29,6 +30,7 @@
'get_available_single_table_synthesizers',
'get_available_multi_table_synthesizers',
'MultiTableUniformSynthesizer',
'ClavaDDPMSynthesizer',
]

for sdv_name in _get_all_sdv_synthesizers():
Expand Down
Loading
Loading