More MCMC steps and seeds

cbg-ethz · Apr 18, 2024 · fa085aa · fa085aa
1 parent e7ca208
commit fa085aa
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/bmi/estimators/external/gmm.py b/src/bmi/estimators/external/gmm.py
@@ -135,8 +135,8 @@ def __init__(
         n_components: int = 10,
         alpha: Optional[float] = None,
         standardize: bool = True,
-        mcmc_num_warmup: int = 500,
-        mcmc_num_samples: int = 500,
+        mcmc_num_warmup: int = 1_000,
+        mcmc_num_samples: int = 1_000,
         mi_estimate_num_samples: int = 1_000,
         n_thinned_samples: Optional[int] = None,
     ) -> None:

diff --git a/workflows/projects/Mixtures/gmm_benchmark.smk b/workflows/projects/Mixtures/gmm_benchmark.smk
@@ -42,7 +42,7 @@ TASKS = [
 TASKS_DICT = {
     task.id: task for task in TASKS
 }
-SEEDS = [0]
+SEEDS = [0, 1, 2]
 
 N_SAMPLES: int = 5_000