minor

aimclub · Dec 7, 2023 · e266264 · e266264
1 parent 46b53a4
commit e266264
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/golem/core/optimisers/adaptive/mab_agents/contextual_mab_agent.py b/golem/core/optimisers/adaptive/mab_agents/contextual_mab_agent.py
@@ -34,7 +34,7 @@ def __init__(self, actions: Sequence[ActType],
         super().__init__(actions=actions, n_jobs=n_jobs, enable_logging=enable_logging,
                          decaying_factor=decaying_factor, is_initial_fit=False)
         self._agent = MAB(arms=self._indices,
-                          learning_policy=LearningPolicy.UCB1(alpha=1.25),
+                          learning_policy=LearningPolicy.EpsilonGreedy(epsilon=0.3),
                           neighborhood_policy=NeighborhoodPolicy.Clusters(),
                           n_jobs=n_jobs)
         self._context_agent = context_agent_type if isinstance(context_agent_type, Callable) else \