RBFKernel

class pmrf.discrepancy_models.kernels.RBFKernel(length_scale: Any = 1.0, *, name: str | None = None)

Bases: CovarianceKernel

Radial Basis Function (Squared Exponential) kernel.

Variables:

length_scale (prx.Parameter) – Characteristic length scale of the correlation (default 1.0).

__call__(x1, x2, key=None)

Evaluate the kernel between two points.

Parameters:

• x1 (jnp.ndarray) – First input point.

• x2 (jnp.ndarray) – Second input point.

• key (jax.random.PRNGKey, optional) – Random key for stochastic kernels.

Returns:

Kernel covariance scalar.

Return type:

jnp.ndarray

length_scale: Parameter = 1.0