From 7d8607c4edb6e0bd37359a57f2938a9286abd00b Mon Sep 17 00:00:00 2001 From: Sebastian Khan Date: Wed, 10 Nov 2021 18:56:11 +0000 Subject: [PATCH] first go at extending the GibbsKernel to work with other base kernels --- src/kernels/gibbskernel.jl | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/kernels/gibbskernel.jl b/src/kernels/gibbskernel.jl index 46e14995d..4011e0cf7 100644 --- a/src/kernels/gibbskernel.jl +++ b/src/kernels/gibbskernel.jl @@ -36,11 +36,11 @@ end GibbsKernel(; lengthscale) = GibbsKernel(lengthscale) -function (k::GibbsKernel)(x, y) +function (k::GibbsKernel)(x, y, base_kernel::Kernel=SqExponentialKernel) lengthscale = k.lengthscale lx = lengthscale(x) ly = lengthscale(y) l = invsqrt2 * hypot(lx, ly) - kernel = (sqrt(lx * ly) / l) * with_lengthscale(SqExponentialKernel(), l) + kernel = (sqrt(lx * ly) / l) * with_lengthscale(base_kernel(), l) return kernel(x, y) end