boax.core.models.kernels.rbf

Contents

boax.core.models.kernels.rbf#

boax.core.models.kernels.rbf(length_scale)#

The Radial basis function (RBF) kernel.

Computes k(x, y) = exp(-||x - y||**2 / (2 * length_scale**2)).

Example

>>> kernel = rbf(jnp.array([0.2, 3.0]))
>>> Kxx = kernel(xs, xs)
Parameters:

length_scale (Union[Array, ndarray, bool, number]) – The parameter controlling how sharp or wide the kernel is.

Return type:

Kernel

Returns:

A RBF Kernel.

Contents