PartialAcquisitionFunction

class baybe.acquisition.PartialAcquisitionFunction

Bases: object

Acquisition function for evaluating points in a hybrid search space.

It can either pin the discrete or the continuous part. The pinned part is assumed to be a tensor of dimension d x 1 where d is the computational dimension of the search space that is to be pinned. The acquisition function is assumed to be defined for the full hybrid space.

Public methods

__init__(acqf, pinned_part, pin_discrete)	Method generated by attrs for class PartialAcquisitionFunction.
set_X_pending(X_pending)	Inform the acquisition function about pending design points.

Public attributes and properties

acqf	The acquisition function for the hybrid space.
pinned_part	The values that will be attached whenever evaluating the acquisition function.
pin_discrete	A flag for denoting whether pinned_part corresponds to the discrete subspace.

__init__(acqf: AcquisitionFunction, pinned_part: Tensor, pin_discrete: bool)

Method generated by attrs for class PartialAcquisitionFunction.

For details on the parameters, see Public attributes and properties.

set_X_pending(X_pending: Tensor | None)

Inform the acquisition function about pending design points.

Enhances the original set_X_pending function from the full acquisition function as we need to store the full point, i.e., the point in the hybrid space for the PartialAcquisitionFunction to work properly.

Parameters:

X_pending (Optional[Tensor]) – n x d Tensor with n d-dim design points that have been submitted for evaluation but have not yet been evaluated.

acqf: AcquisitionFunction

The acquisition function for the hybrid space.

pin_discrete: bool

A flag for denoting whether pinned_part corresponds to the discrete subspace.

pinned_part: Tensor

The values that will be attached whenever evaluating the acquisition function.