removed search space from algorithm class and fixed hyperparameter na…

…ming
automl · Jun 5, 2024 · ed05c99 · ed05c99
1 parent bb0c6b6
commit ed05c99
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 13 deletions.
diff --git a/arlbench/core/algorithms/algorithm.py b/arlbench/core/algorithms/algorithm.py
@@ -97,18 +97,6 @@ def get_hpo_config_space(seed: int | None = None) -> ConfigurationSpace:
             ConfigurationSpace: Hyperparameter configuration space of the algorithm.
         """
 
-    @staticmethod
-    @abstractmethod
-    def get_hpo_search_space(seed: int | None = None) -> ConfigurationSpace:
-        """Returns the hyperparameter search space of the algorithm. However, this can be adapted to fit a given HPO method.
-
-        Args:
-            seed (int | None, optional): Random generator seed that is used to sample configurations. Defaults to None.
-
-        Returns:
-            ConfigurationSpace: Hyperparameter search space of the algorithm.
-        """
-
     @staticmethod
     @abstractmethod
     def get_default_hpo_config() -> Configuration:

diff --git a/arlbench/core/algorithms/dqn/dqn.py b/arlbench/core/algorithms/dqn/dqn.py
@@ -223,7 +223,7 @@ def get_hpo_config_space(seed: int | None = None) -> ConfigurationSpace:
                 "tau": Float("tau", (0.01, 1.0), default=1.0),
                 "initial_epsilon": Float("initial_epsilon", (0.5, 1.0), default=1.0),
                 "target_epsilon": Float("target_epsilon", (0.001, 0.2), default=0.05),
-                "exploration_fraction": Float("initial_epsilon", (0.005, 0.5), default=0.1),
+                "exploration_fraction": Float("exploration_fraction", (0.005, 0.5), default=0.1),
                 "use_target_network": Categorical(
                     "use_target_network", [True, False], default=True
                 ),