Add BayesianGenerator (Gaussian Copula)

tabgan/auto_synth.py

@@ -26,6 +26,7 @@ from .sampler import (
     GANGenerator,
     ForestDiffusionGenerator,
     OriginalGenerator,
+    BayesianGenerator,
 )
 
 logger = logging.getLogger(__name__)
@@ -67,6 +68,10 @@ _DEFAULT_GENERATORS = {
         "cls": ForestDiffusionGenerator,
         "params": {},
     },
+    "Bayesian (Copula)": {
+        "cls": BayesianGenerator,
+        "params": {},
+    },
     "Random Baseline": {
         "cls": OriginalGenerator,
         "params": {},