Skip to content

Search Space

AbstractSearchSpace dataclass

AbstractSearchSpace()

Bases: ABC

The AbstractSearchSpace class is an abstract base class for search spaces, which are used to define domains for sampling and optimisation functionality in GPJax.

dimensionality abstractmethod property

dimensionality

Dimensionality of the search space. Returns: int: Dimensionality of the search space.

sample abstractmethod

sample(num_points, key)

Sample points from the search space. Args: num_points (int): Number of points to be sampled from the search space. key (KeyArray): JAX PRNG key. Returns: Float[Array, "N D"]: num_points points sampled from the search space.

ContinuousSearchSpace dataclass

ContinuousSearchSpace(lower_bounds, upper_bounds)

Bases: AbstractSearchSpace

The ContinuousSearchSpace class is used to bound the domain of continuous real functions of dimension \(D\).

sample

sample(num_points, key)

Sample points from the search space using a Halton sequence.

Parameters:

  • num_points (int) –

    Number of points to be sampled from the search space.

  • key (KeyArray) –

    JAX PRNG key.

Returns: Float[Array, "N D"]: num_points points sampled using the Halton sequence from the search space.