Update _catcma.py

CyberAgentAILab · Aug 13, 2024 · fdc7461 · fdc7461
1 parent 7513df8
commit fdc7461
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cmaes/_catcma.py b/cmaes/_catcma.py
@@ -117,7 +117,7 @@ def __init__(
         # (CMA uses negative weights while CatCMA uses positive weights.)
         weights_prime = np.array(
             [
-                math.log((population_size + 1) / 2) - math.log(i + 1) if i <= mu else 0
+                math.log((population_size + 1) / 2) - math.log(i + 1) if i < mu else 0
                 for i in range(population_size)
             ]
         )