Experiments notebook adn conf different kernel comparison

mackelab · Jun 4, 2024 · 742ae09 · 742ae09
1 parent 8fdb0b8
commit 742ae09
Show file tree

Hide file tree

Showing 2 changed files with 120 additions and 22 deletions.
diff --git a/configs/conf_mmd_main_scaling_experiment_different_kernels.yaml b/configs/conf_mmd_main_scaling_experiment_different_kernels.yaml
@@ -5,18 +5,18 @@ data: ["toy_2d" , "random", "random"]
 augmentation: ['gauss', 'one_dim_shift', 'one_dim_shift',]
 
 # number of samples and dimensions
-n:  [1000,1000,1000] #[10000, 10000, 10000] #samples Note that for main figure 10k
+n:  [10000,10000,10000] #[10000, 10000, 10000] #samples Note that for main figure 10k
 d: [2, 10, 1000] # dimensions
 
-mmd_bandwidth: [[1, 5, 10],[1, 5, 10],[1, 5, 10]]
+mmd_bandwidth:  [1,5,10] # only the rbf has the bandwidth parameter
 
 # sample size experiments
-experiments: ["ScaleSampleSizeMMD", "ScaleSampleSizeMMD","ScaleSampleSizeMMD"] 
+experiments: ["ScaleSampleSizeMMD", "ScaleSampleSizeMMDlinear","ScaleSampleSizeMMDenergy"] 
 sample_size: [50, 100, 200, 500, 1000, 2000, 3000, 4000]
 runs: 5 # number of sample selection for errorbars
 
 # dimensionality experiments
-experiments_dim: ["ScaleDimMMD", "ScaleDimMMD", "ScaleDimMMD"]
+experiments_dim: ["ScaleDimMMD", "ScaleDimMMDlinear", "ScaleDimMMDenergy"]
 dim_sizes: [5, 10, 50, 100, 500, 1000]
 runs_dim: 5 # number of sample selection for errorbars
 

diff --git a/docs/notebooks/mmd/MMD_scaling_experiment_different_kernels.ipynb b/docs/notebooks/mmd/MMD_scaling_experiment_different_kernels.ipynb