Franzi Review MMD Notebooks

mackelab · Mar 8, 2024 · 94e4351 · 94e4351
1 parent 10e4270
commit 94e4351
Show file tree

Hide file tree

Showing 5 changed files with 60 additions and 66 deletions.
diff --git a/configs/conf_main_scaling_experiment.yaml b/configs/conf_main_scaling_experiment.yaml
@@ -8,6 +8,7 @@ augmentation: ['gauss', 'one_dim_shift', 'one_dim_shift',]
 n: [10000,10000,10000] #samples 
 d: [2, 10, 1000] # dimensions
 
+# MMD bandwith parameter
 mmd_bandwidth: [[1,1,1], [5,5,5], [10,10,10]]
 
 # sample size experiments
@@ -23,7 +24,6 @@ runs_dim: 5 # number of sample selection for errorbars
 # seed for reproducibility
 seed: 0
 
-
 # for the reduced sample size experiments
 #sample_size: [8, 10, 20, 50, 80]
 #n:  [500, 500, 500] 
diff --git a/docs/notebooks/comparison/main_scaling_experiment.ipynb b/docs/notebooks/comparison/main_scaling_experiment.ipynb
@@ -4,7 +4,7 @@
    "cell_type": "markdown",
    "metadata": {},
    "source": [
-    "## Perform main sample size and dimensionality scaling experiment"
+    "# Perform main sample size and dimensionality scaling experiment"
    ]
   },
   {
@@ -21,7 +21,7 @@
    "cell_type": "markdown",
    "metadata": {},
    "source": [
-    "### Load the configs and set up the plotting "
+    "## Load the configs and set up the plotting "
    ]
   },
   {

diff --git a/docs/notebooks/mmd/MMD_MainFig.ipynb b/docs/notebooks/mmd/MMD_MainFig.ipynb
diff --git a/docs/notebooks/mmd/MMD_bandwidth_experiment.ipynb b/docs/notebooks/mmd/MMD_bandwidth_experiment.ipynb
@@ -42,8 +42,6 @@
     "# inline plotting\n",
     "%matplotlib inline\n",
     "\n",
-    "\n",
-    "\n",
     "print(\"Running experiments...\")\n",
     "# load the config file\n",
     "cfg = get_cfg_from_file(\"conf_mmd_bandwidth_experiment\")\n",
@@ -59,14 +57,12 @@
     "assert len(cfg.data) == len(cfg.n) == len(cfg.d), \"Data, n and d must be lists of the same length\"\n",
     "    \n",
     "# setup colors and labels for plotting\n",
-    "\n",
     "color_dict = {\"wasserstein\": \"#cc241d\",\n",
     "              \"mmd\": \"#eebd35\",\n",
     "              \"c2st\": \"#458588\",\n",
     "              \"fid\": \"#8ec07c\", \n",
     "              \"kl\": \"#8ec07c\"}\n",
     "\n",
-    "\n",
     "col_map = {'ScaleSampleSizeKL':'kl', 'ScaleSampleSizeSW':'wasserstein',\n",
     "           'ScaleSampleSizeMMD':'mmd', 'ScaleSampleSizeC2ST':'c2st',\n",
     "           'ScaleSampleSizeFID':'fid', 'ScaleDimKL':'kl', 'ScaleDimSW':'wasserstein',\n",
@@ -96,7 +92,7 @@
     "    \n",
     "label_list = [label_true, label_shift]\n",
     "label_list[1]['toy_2d'] = 'approx.'\n",
-    "label_list[1]['random'] = 'shifted'\n"
+    "label_list[1]['random'] = 'shifted'"
    ]
   },
   {

diff --git a/docs/notebooks/mmd/MMD_scaling_experiment_different_bandwidths.ipynb b/docs/notebooks/mmd/MMD_scaling_experiment_different_bandwidths.ipynb
@@ -4,7 +4,7 @@
    "cell_type": "markdown",
    "metadata": {},
    "source": [
-    "## Test MMD bandwidth sensitivity in all scaling experiments"
+    "# Test MMD bandwidth sensitivity in all scaling experiments"
    ]
   },
   {
@@ -21,7 +21,7 @@
    "cell_type": "markdown",
    "metadata": {},
    "source": [
-    "### Load the configs and set up the plotting "
+    "## Load the configs and set up the plotting "
    ]
   },
   {
@@ -99,7 +99,7 @@
     "    \n",
     "label_list = [label_true, label_shift]\n",
     "label_list[1]['toy_2d'] = 'approx.'\n",
-    "label_list[1]['random'] = 'shifted'\n"
+    "label_list[1]['random'] = 'shifted'"
    ]
   },
   {