|
34 | 34 | from ax.modelbridge.random import RandomModelBridge
|
35 | 35 | from ax.modelbridge.torch import TorchModelBridge
|
36 | 36 | from ax.modelbridge.transforms.base import Transform
|
37 |
| -from ax.modelbridge.transforms.centered_unit_x import CenteredUnitX |
38 | 37 | from ax.modelbridge.transforms.choice_encode import (
|
39 | 38 | ChoiceToNumericChoice,
|
40 | 39 | OrderedChoiceToIntegerRange,
|
|
59 | 58 | from ax.models.discrete.eb_thompson import EmpiricalBayesThompsonSampler
|
60 | 59 | from ax.models.discrete.full_factorial import FullFactorialGenerator
|
61 | 60 | from ax.models.discrete.thompson import ThompsonSampler
|
62 |
| -from ax.models.random.alebo_initializer import ALEBOInitializer |
63 | 61 | from ax.models.random.sobol import SobolGenerator
|
64 | 62 | from ax.models.random.uniform import UniformGenerator
|
65 |
| -from ax.models.torch.alebo import ALEBO |
66 | 63 | from ax.models.torch.botorch import BotorchModel
|
67 | 64 | from ax.models.torch.botorch_modular.model import (
|
68 | 65 | BoTorchModel as ModularBoTorchModel,
|
|
137 | 134 | TaskChoiceToIntTaskChoice,
|
138 | 135 | ]
|
139 | 136 |
|
140 |
| -ALEBO_X_trans: List[Type[Transform]] = [RemoveFixed, IntToFloat, CenteredUnitX] |
141 |
| -ALEBO_Y_trans: List[Type[Transform]] = [Derelativize, StandardizeY] |
142 |
| - |
143 | 137 | STANDARD_TORCH_BRIDGE_KWARGS: Dict[str, Any] = {"torch_dtype": torch.double}
|
144 | 138 |
|
145 | 139 |
|
@@ -218,17 +212,6 @@ class ModelSetup(NamedTuple):
|
218 | 212 | transforms=ST_MTGP_trans,
|
219 | 213 | standard_bridge_kwargs=STANDARD_TORCH_BRIDGE_KWARGS,
|
220 | 214 | ),
|
221 |
| - "ALEBO": ModelSetup( |
222 |
| - bridge_class=TorchModelBridge, |
223 |
| - model_class=ALEBO, |
224 |
| - transforms=ALEBO_X_trans + ALEBO_Y_trans, |
225 |
| - standard_bridge_kwargs=STANDARD_TORCH_BRIDGE_KWARGS, |
226 |
| - ), |
227 |
| - "ALEBO_Initializer": ModelSetup( |
228 |
| - bridge_class=RandomModelBridge, |
229 |
| - model_class=ALEBOInitializer, |
230 |
| - transforms=ALEBO_X_trans, |
231 |
| - ), |
232 | 215 | "BO_MIXED": ModelSetup(
|
233 | 216 | bridge_class=TorchModelBridge,
|
234 | 217 | model_class=ModularBoTorchModel,
|
@@ -444,10 +427,8 @@ class Models(ModelRegistryBase):
|
444 | 427 | MOO = "MOO"
|
445 | 428 | ST_MTGP_LEGACY = "ST_MTGP_LEGACY"
|
446 | 429 | ST_MTGP = "ST_MTGP"
|
447 |
| - ALEBO = "ALEBO" |
448 | 430 | BO_MIXED = "BO_MIXED"
|
449 | 431 | ST_MTGP_NEHVI = "ST_MTGP_NEHVI"
|
450 |
| - ALEBO_INITIALIZER = "ALEBO_Initializer" |
451 | 432 | CONTEXT_SACBO = "Contextual_SACBO"
|
452 | 433 |
|
453 | 434 | @classmethod
|
|
0 commit comments