custom#

Custom implementations of acquisition functions using BoTorch API

Classes

`qMean`(model[, sampler, objective, ...])	Acquisition function which optimizes for the maximum value of the posterior mean
`qSpaceFill`(model, X_baseline[, sampler, ...])	Acquisition function which optimizes for the maximum value of minimum distance between a point and the training data

class obsidian.acquisition.custom.qMean(model: Model, sampler: MCSampler | None = None, objective: MCAcquisitionObjective | None = None, posterior_transform: PosteriorTransform | None = None, X_pending: Tensor | None = None)[source]#

Bases: MCAcquisitionFunction

Acquisition function which optimizes for the maximum value of the posterior mean

forward(x: Tensor) → Tensor[source]#: Evaluate the acquisition function on the candidate set x

class obsidian.acquisition.custom.qSpaceFill(model: Model, X_baseline: Tensor, sampler: MCSampler | None = None, objective: MCAcquisitionObjective | None = None, posterior_transform: PosteriorTransform | None = None, X_pending: Tensor | None = None)[source]#

Bases: MCAcquisitionFunction

Acquisition function which optimizes for the maximum value of minimum distance between a point and the training data

forward(x: Tensor) → Tensor[source]#: Evaluate the acquisition function on the candidate set x