feat: add smac sweeper basis

automl · Feb 13, 2024 · 7b21955 · 7b21955
1 parent 79e45a0
commit 7b21955
Show file tree

Hide file tree

Showing 5 changed files with 617 additions and 0 deletions.
diff --git a/hydra_plugins/hypersweeper/smac_sweeper/__init__.py b/hydra_plugins/hypersweeper/smac_sweeper/__init__.py
diff --git a/hydra_plugins/hypersweeper/smac_sweeper/config.py b/hydra_plugins/hypersweeper/smac_sweeper/config.py
@@ -0,0 +1,27 @@
+from __future__ import annotations
+
+from dataclasses import dataclass, field
+from typing import Any, Dict, Optional
+
+from hydra.core.config_store import ConfigStore
+
+
+@dataclass
+class SMACSweeperConfig:
+    _target_: str = "hydra_plugins.hydra_smac_sweeper.smac_sweeper.SMACSweeper"
+    search_space: Optional[Dict] = field(default_factory=dict)
+    resume: Optional[str] = None
+    optimizer: Optional[str] = "smac"
+    budget: Optional[Any] = None
+    budget_variable: Optional[str] = None
+    loading_variable: Optional[str] = None
+    saving_variable: Optional[str] = None
+    smac_kwargs: Optional[Dict] = field(default_factory=dict)
+
+
+ConfigStore.instance().store(
+    group="hydra/sweeper",
+    name="SMAC",
+    node=SMACSweeperConfig,
+    provider="hydra_smac_sweeper",
+)