himalaya.kernel_ridge.rbf_kernel

himalaya.kernel_ridge.rbf_kernel(X, Y=None, gamma=None)

Compute the rbf (gaussian) kernel between X and Y:

K(x, y) = exp(-gamma ||x-y||^2)

for each pair of rows x in X and y in Y.

Parameters

Xarray of shape (n_samples_X, n_features)

Train or test features.

Yarray of shape (n_samples_Y, n_features)

Train features.

gammafloat, default None

If None, defaults to 1.0 / n_features

Returns

Karray of shape (n_samples_X, n_samples_Y)

Computed kernel.