mxfusion.components.distributions.gp.kernels.stationary

Members

class mxfusion.components.distributions.gp.kernels.stationary.StationaryKernel(input_dim, ARD=False, variance=1.0, lengthscale=1.0, name='stationary', active_dims=None, dtype=None, ctx=None)

Bases: mxfusion.components.distributions.gp.kernels.kernel.NativeKernel

The base class for Stationary kernels (covariance functions).

Stationary kernels (covariance functions). Stationary covariance fucntion depend only on r^2, where r^2 is defined as .. math:

r2(x, x') = \sum_{q=1}^Q (x_q - x'_q)^2

The covariance function k(x, x’ can then be written k(r).

In this implementation, r is scaled by the lengthscales parameter(s): .. math:

r2(x, x') = \sum_{q=1}^Q \frac{(x_q - x'_q)^2}{\ell_q^2}.

By default, there’s only one lengthscale: seaprate lengthscales for each dimension can be enables by setting ARD=True.

Parameters:
  • input_dim (int) – the number of dimensions of the kernel. (The total number of active dimensions).
  • ARD (boolean) – a binary switch for Automatic Relevance Determination (ARD). If true, the squared distance is divided by a lengthscale for individual dimensions.
  • variance (float or MXNet NDArray) – the initial value for the variance parameter (scalar), which scales the whole covariance matrix.
  • lengthscale (float or MXNet NDArray) – the initial value for the lengthscale parameter.
  • name (str) – the name of the kernel. The name is used to access kernel parameters.
  • active_dims ([int] or None) – The dimensions of the inputs that are taken for the covariance matrix computation. (default: None, taking all the dimensions).
  • dtype (numpy.float32 or numpy.float64) – the data type for float point numbers.
  • ctx (None or mxnet.cpu or mxnet.gpu) – the mxnet context (default: None/current context).