Skip to content

Commit 8b27bc4

Browse files
removed duplicate PyTorch example acquisition function
1 parent e06d57e commit 8b27bc4

File tree

1 file changed

+0
-14
lines changed

1 file changed

+0
-14
lines changed

elfi/methods/bo/botorch_acquisition.py

-14
Original file line numberDiff line numberDiff line change
@@ -132,17 +132,3 @@ def acquire(self, n, t=None, std_scale=None):
132132
)
133133
candidate = get_best_candidates(batch_candidates, -batch_acq_values)
134134
return candidate.detach().numpy()
135-
136-
137-
class SOBERUCB:
138-
def __init__(self, model, label="UCB", sample_size=1, exploration_rate=10):
139-
self.label = label
140-
self.exploration_rate = exploration_rate
141-
self.beta = (2 * log(
142-
sample_size**(2 * model.dim + 2) * pi**2
143-
/ (3 / self.exploration_rate)
144-
))
145-
self.af = UpperConfidenceBound(model, beta=self.beta)
146-
147-
def __call__(self, x):
148-
return self.af(x.unsqueeze(1)).detach()

0 commit comments

Comments
 (0)