Use bernoully distribution to select samples

Ekumen-OS · Nov 11, 2022 · a38afc6 · a38afc6
1 parent 350210a
commit a38afc6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/beluga/core/include/beluga/algorithm/sampling.h b/beluga/core/include/beluga/algorithm/sampling.h
@@ -12,8 +12,8 @@ namespace beluga {
 
 template <class Function1, class Function2, class RandomNumberGenerator>
 auto random_select(Function1&& first, Function2&& second, RandomNumberGenerator& generator, double probability) {
-  return [&, probability, distribution = std::uniform_real_distribution{}]() mutable {
-    return distribution(generator) < probability ? first() : second();
+  return [&, distribution = std::bernoulli_distribution{probability}]() mutable {
+    return distribution(generator) ? first() : second();
   };
 }