Skip to content

Commit 933756b

Browse files
saitcakmakfacebook-github-bot
authored andcommitted
Remove REMBO & ALEBO (#2458)
Summary: Pull Request resolved: #2458 These were marked for removal in v0.3.9. Last release was v0.4.0. For users interested in BO with high-dimensional inputs, we recommend using SAASBO instead. See https://ax.dev/tutorials/saasbo.html for a tutorial Reviewed By: bletham Differential Revision: D57360740 fbshipit-source-id: d8214b935c9e22c9d690881d12897558742392b2
1 parent 6092ac3 commit 933756b

17 files changed

+0
-2760
lines changed

ax/modelbridge/registry.py

-19
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,6 @@
3434
from ax.modelbridge.random import RandomModelBridge
3535
from ax.modelbridge.torch import TorchModelBridge
3636
from ax.modelbridge.transforms.base import Transform
37-
from ax.modelbridge.transforms.centered_unit_x import CenteredUnitX
3837
from ax.modelbridge.transforms.choice_encode import (
3938
ChoiceToNumericChoice,
4039
OrderedChoiceToIntegerRange,
@@ -59,10 +58,8 @@
5958
from ax.models.discrete.eb_thompson import EmpiricalBayesThompsonSampler
6059
from ax.models.discrete.full_factorial import FullFactorialGenerator
6160
from ax.models.discrete.thompson import ThompsonSampler
62-
from ax.models.random.alebo_initializer import ALEBOInitializer
6361
from ax.models.random.sobol import SobolGenerator
6462
from ax.models.random.uniform import UniformGenerator
65-
from ax.models.torch.alebo import ALEBO
6663
from ax.models.torch.botorch import BotorchModel
6764
from ax.models.torch.botorch_modular.model import (
6865
BoTorchModel as ModularBoTorchModel,
@@ -137,9 +134,6 @@
137134
TaskChoiceToIntTaskChoice,
138135
]
139136

140-
ALEBO_X_trans: List[Type[Transform]] = [RemoveFixed, IntToFloat, CenteredUnitX]
141-
ALEBO_Y_trans: List[Type[Transform]] = [Derelativize, StandardizeY]
142-
143137
STANDARD_TORCH_BRIDGE_KWARGS: Dict[str, Any] = {"torch_dtype": torch.double}
144138

145139

@@ -218,17 +212,6 @@ class ModelSetup(NamedTuple):
218212
transforms=ST_MTGP_trans,
219213
standard_bridge_kwargs=STANDARD_TORCH_BRIDGE_KWARGS,
220214
),
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-
),
232215
"BO_MIXED": ModelSetup(
233216
bridge_class=TorchModelBridge,
234217
model_class=ModularBoTorchModel,
@@ -444,10 +427,8 @@ class Models(ModelRegistryBase):
444427
MOO = "MOO"
445428
ST_MTGP_LEGACY = "ST_MTGP_LEGACY"
446429
ST_MTGP = "ST_MTGP"
447-
ALEBO = "ALEBO"
448430
BO_MIXED = "BO_MIXED"
449431
ST_MTGP_NEHVI = "ST_MTGP_NEHVI"
450-
ALEBO_INITIALIZER = "ALEBO_Initializer"
451432
CONTEXT_SACBO = "Contextual_SACBO"
452433

453434
@classmethod

ax/modelbridge/strategies/__init__.py

-5
This file was deleted.

ax/modelbridge/strategies/alebo.py

-140
This file was deleted.

0 commit comments

Comments
 (0)