grf_gp.kernels.general module

class grf_gp.kernels.general.GeneralGRFKernel(rw_mats, max_walk_length: int, **kwargs)

Bases: BaseGRFKernel

Learnable GRF kernel with an unconstrained modulation function.

property modulation_function: Tensor

Return the learnable modulation coefficients.

Returns:

Unconstrained modulation vector indexed by walk length.

class grf_gp.kernels.general.GeneralLowRankGRFKernel(rw_mats, max_walk_length: int, proj_dim: int, jlt_seed: int = 42, **kwargs)

Bases: LowRankGRFKernel

Low-rank GRF kernel with an unconstrained modulation function.

property modulation_function: Tensor

Return the learnable modulation coefficients.

Returns:

Unconstrained modulation vector indexed by walk length.