Add CopulaGAN Model (#202)

* Add CopulaGAN model * Add control over cuda device * Curate docstrings and imports * Add integration test for copulagan * Update docs and tutorials * Fix lint
sdv-dev · Oct 6, 2020 · bccbe9f · bccbe9f
1 parent 7e625f0
commit bccbe9f
Show file tree

Hide file tree

Showing 20 changed files with 5,890 additions and 1,906 deletions.
diff --git a/docs/api_reference/tabular.rst b/docs/api_reference/tabular.rst
@@ -17,6 +17,8 @@ GaussianCopula
    GaussianCopula.get_metadata
    GaussianCopula.get_parameters
    GaussianCopula.set_parameters
+   GaussianCopula.get_distributions
+   GaussianCopula.get_likelihood
    GaussianCopula.save
    GaussianCopula.load
 
@@ -32,3 +34,17 @@ CTGAN
    CTGAN.get_metadata
    CTGAN.save
    CTGAN.load
+
+CTGAN
+~~~~~
+
+.. autosummary::
+   :toctree: api/
+
+   CopulaGAN
+   CopulaGAN.fit
+   CopulaGAN.sample
+   CopulaGAN.get_metadata
+   CopulaGAN.get_distributions
+   CopulaGAN.save
+   CopulaGAN.load