Merge pull request #164 from ryota717/refactor-mab

small refactor in `mab_epsilon_greedy`
optuna · Oct 15, 2024 · 7892786 · 7892786
2 parents f806fbb + e4ad3bb
commit 7892786
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/package/samplers/mab_epsilon_greedy/mab_epsilon_greedy.py b/package/samplers/mab_epsilon_greedy/mab_epsilon_greedy.py
@@ -62,10 +62,10 @@ def sample_independent(
             if study.direction == StudyDirection.MINIMIZE:
                 return min(
                     param_distribution.choices,
-                    key=lambda x: rewards_by_choice[x] / max(cnt_by_choice[x], 1),
+                    key=lambda x: rewards_by_choice[x] / cnt_by_choice[x],
                 )
             else:
                 return max(
                     param_distribution.choices,
-                    key=lambda x: rewards_by_choice[x] / max(cnt_by_choice[x], 1),
+                    key=lambda x: rewards_by_choice[x] / cnt_by_choice[x],
                 )