fixed detectron conditions

MEDomics-UdeS · Aug 31, 2024 · 809763e · 809763e
1 parent 453e79e
commit 809763e
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/MED3pa/datasets/masked.py b/MED3pa/datasets/masked.py
@@ -156,6 +156,7 @@ def sample_random(self, N: int, seed: int) -> 'MaskedDataset':
         # Set the seed for reproducibility and generate random indices
         rng = np.random.RandomState(seed)
         random_indices = rng.permutation(len(self.__observations))[:N]
+        self.__sample_counts[random_indices] += 1
 
         # Extract the sampled observations and labels
         sampled_obs = self.__observations[random_indices, :]

diff --git a/MED3pa/detectron/experiment.py b/MED3pa/detectron/experiment.py
@@ -248,7 +248,8 @@ def run(datasets: DatasetsManager,
             'num_runs': num_calibration_runs,
             'patience': patience,
             'allow_margin': allow_margin,
-            'margin': margin
+            'margin': margin,
+            'sampling-method':sampling,
         }
         experiment_config = {
             'experiment_name': "DetectronExperiment",