Hotfix RUSBoost replacement hp value space.

AxiomAlive · Feb 20, 2025 · 6083e8d · 6083e8d
1 parent 8f0ea5e
commit 6083e8d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/config_spaces/ensemble/boost.py b/config_spaces/ensemble/boost.py
@@ -51,7 +51,7 @@ def generate_algorithm_configuration_space(cls, model_class=None):
 
 
 class RUSBoostGenerator(AdaGenerator):
-    replacement = hp.choice("rus_boost.replacement", [True, False])
+    replacement = False
 
     @classmethod
     def generate_algorithm_configuration_space(cls, model_class=None):

diff --git a/experiment/imba.py b/experiment/imba.py
@@ -37,6 +37,8 @@ def trainable(config):
         ray.train.report(trial_result)
 
 
+#TODO: use ray.tune.Trainable directly
+#
 # class RayTrainable(ray.tune.Trainable):
 #     def setup(self, config):
 #         self.algorithm_configuration = config["algorithm_configuration"]