Update tutorial

vanderschaarlab · Aug 7, 2023 · 8831220 · 8831220
1 parent ec9824c
commit 8831220
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/tutorials/tutorial1_add_a_new_plugin.ipynb b/tutorials/tutorial1_add_a_new_plugin.ipynb
@@ -204,6 +204,16 @@
     "__Note__ : `synthcity` also includes a dedicated CTGAN re-implementation."
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "2ec31e38",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "!pip install ctgan"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": null,
@@ -217,7 +227,7 @@
     "# third party\n",
     "import numpy as np\n",
     "import pandas as pd\n",
-    "from ctgan import CTGANSynthesizer\n",
+    "from ctgan import CTGAN\n",
     "\n",
     "# synthcity absolute\n",
     "from synthcity.plugins.core.dataloader import DataLoader, GenericDataLoader\n",
@@ -243,7 +253,7 @@
     "    ) -> None:\n",
     "        super().__init__(**kwargs)\n",
     "        self.cat_limit = cat_limit\n",
-    "        self.model = CTGANSynthesizer(\n",
+    "        self.model = CTGAN(\n",
     "            embedding_dim=embedding_n_units,\n",
     "            batch_size=batch_size,\n",
     "            epochs=n_iter,\n",