Removed obsolete code and tests

Author: theogf

src/kernels/fbm.jl

@@ -10,27 +10,27 @@ For `h=1/2`, this is the Wiener Kernel, for `h>1/2`, the increments are
 positively correlated and for `h<1/2` the increments are negatively correlated.
 """
 struct FBMKernel{T<:Real} <: BaseKernel
-    h::T
+    h::Vector{T}
     function FBMKernel(; h::T=0.5) where {T<:Real}
         @assert h<=1.0 && h>=0.0 "FBMKernel: Given Hurst index h is invalid."
-        return new{T}(h)
+        return new{T}([h])
     end
 end
 
-Base.show(io::IO, κ::FBMKernel) = print(io, "Fractional Brownian Motion Kernel (h = $(k.h))")
+Base.show(io::IO, κ::FBMKernel) = print(io, "Fractional Brownian Motion Kernel (h = $(first(k.h)))")
 
 _fbm(modX, modY, modXY, h) = (modX^h + modY^h - modXY^h)/2
 
 function kernelmatrix(κ::FBMKernel, X::AbstractMatrix; obsdim::Int = defaultobs)
     @assert obsdim ∈ [1,2] "obsdim should be 1 or 2 (see docs of kernelmatrix))"
-    modX = sum(abs2, X; dims = 3 - obsdim)
+    modX = sum(abs2, X; dims = feature_dim(obsdim))
     modXX = pairwise(SqEuclidean(), X, dims = obsdim)
     return _fbm.(vec(modX), reshape(modX, 1, :), modXX, κ.h)
 end
 
 function kernelmatrix!(K::AbstractMatrix, κ::FBMKernel, X::AbstractMatrix; obsdim::Int = defaultobs)
     @assert obsdim ∈ [1,2] "obsdim should be 1 or 2 (see docs of kernelmatrix))"
-    modX = sum(abs2, X; dims = 3 - obsdim)
+    modX = sum(abs2, X; dims = feature_dim(obsdim))
     modXX = pairwise(SqEuclidean(), X, dims = obsdim)
     K .= _fbm.(vec(modX), reshape(modX, 1, :), modXX, κ.h)
     return K
@@ -43,9 +43,9 @@ function kernelmatrix(
     obsdim::Int = defaultobs,
 )
     @assert obsdim ∈ [1,2] "obsdim should be 1 or 2 (see docs of kernelmatrix))"
-    modX = sum(abs2, X, dims=3-obsdim)
-    modY = sum(abs2, Y, dims=3-obsdim)
-    modXY = pairwise(SqEuclidean(), X, Y,dims=obsdim)
+    modX = sum(abs2, X, dims = feature_dim(obsdim))
+    modY = sum(abs2, Y, dims = feature_dim(obsdim))
+    modXY = pairwise(SqEuclidean(), X, Y,dims = obsdim)
     return _fbm.(vec(modX), reshape(modY, 1, :), modXY, κ.h)
 end
 
@@ -57,9 +57,9 @@ function kernelmatrix!(
     obsdim::Int = defaultobs,
 )
     @assert obsdim ∈ [1,2] "obsdim should be 1 or 2 (see docs of kernelmatrix))"
-    modX = sum(abs2, X, dims=3-obsdim)
-    modY = sum(abs2, Y, dims=3-obsdim)
-    modXY = pairwise(SqEuclidean(), X, Y,dims=obsdim)
+    modX = sum(abs2, X, dims = feature_dim(obsdim))
+    modY = sum(abs2, Y, dims = feature_dim(obsdim))
+    modXY = pairwise(SqEuclidean(), X, Y,dims = obsdim)
     K .= _fbm.(vec(modX), reshape(modY, 1, :), modXY, κ.h)
     return K
 end
@@ -72,23 +72,15 @@ function _kernel(
         obsdim::Int = defaultobs
     )
     @assert length(x) == length(y) "x and y don't have the same dimension!"
-    return κ(x,y)
+    return kappa(κ, x, y)
 end
 
-#Syntactic Sugar
-function (κ::FBMKernel)(x::AbstractVector{<:Real}, y::AbstractVector{<:Real})
+function kappa(κ::FBMKernel, x::AbstractVector{<:Real}, y::AbstractVector{<:Real})
     modX = sum(abs2, x)
     modY = sum(abs2, y)
     modXY = sqeuclidean(x, y)
-    return (modX^κ.h + modY^κ.h - modXY^κ.h)/2
+    h = first(κ.h)
+    return (modX^h + modY^h - modXY^h)/2
 end
 
-(κ::FBMKernel)(x::Real, y::Real) = (abs2(x)^κ.h + abs2(y)^κ.h - abs2(x-y)^κ.h)/2
-
-function (κ::FBMKernel)(X::AbstractMatrix{<:Real}, Y::AbstractMatrix{<:Real}; obsdim::Integer=defaultobs)
-    return kernelmatrix(κ, X, Y, obsdim=obsdim)
-end
-
-function (κ::FBMKernel)(X::AbstractMatrix{<:Real}; obsdim::Integer=defaultobs)
-    return kernelmatrix(κ, X, obsdim=obsdim)
-end
+(κ::FBMKernel)(x::Real, y::Real) = (abs2(x)^first(κ.h) + abs2(y)^first(κ.h) - abs2(x-y)^first(κ.h))/2

src/matrix/kernelmatrix.jl

@@ -1,9 +1,8 @@
 """
-```
-    kernelmatrix!(K::Matrix, κ::Kernel, X::Matrix; obsdim::Integer=2)
-    kernelmatrix!(K::Matrix, κ::Kernel, X::Matrix, Y::Matrix; obsdim::Integer=2)
-```
-In-place version of `kernelmatrix` where pre-allocated matrix `K` will be overwritten with the kernel matrix.
+    kernelmatrix!(K::Matrix, κ::Kernel, X::Matrix; obsdim::Integer = 2)
+    kernelmatrix!(K::Matrix, κ::Kernel, X::Matrix, Y::Matrix; obsdim::Integer = 2)
+
+In-place version of [`kernelmatrix`](@ref) where pre-allocated matrix `K` will be overwritten with the kernel matrix.
 """
 kernelmatrix!
 
@@ -21,7 +20,7 @@ function kernelmatrix!(
         map!(x->kappa(κ,x),K,pairwise(metric(κ),X,dims=obsdim))
 end
 
-kernelmatrix!(K::Matrix, κ::TransformedKernel, X::AbstractMatrix; obsdim::Int = defaultobs) =
+kernelmatrix!(K::AbstractMatrix, κ::TransformedKernel, X::AbstractMatrix; obsdim::Int = defaultobs) =
         kernelmatrix!(K, kernel(κ), apply(κ.transform, X, obsdim = obsdim), obsdim = obsdim)
 
 function kernelmatrix!(
@@ -61,13 +60,12 @@ _kernel(κ::TransformedKernel, x::AbstractVector, y::AbstractVector; obsdim::Int
         _kernel(kernel(κ), apply(κ.transform, x), apply(κ.transform, y), obsdim = obsdim)
 
 """
-```
-    kernelmatrix(κ::Kernel, X::Matrix ; obsdim::Int=2)
-    kernelmatrix(κ::Kernel, X::Matrix, Y::Matrix; obsdim::Int=2)
-```
+    kernelmatrix(κ::Kernel, X::Matrix; obsdim::Int = 2)
+    kernelmatrix(κ::Kernel, X::Matrix, Y::Matrix; obsdim::Int = 2)
+
 Calculate the kernel matrix of `X` (and `Y`) with respect to kernel `κ`.
-`obsdim=1` means the matrix `X` (and `Y`) has size #samples x #dimension
-`obsdim=2` means the matrix `X` (and `Y`) has size #dimension x #samples
+`obsdim = 1` means the matrix `X` (and `Y`) has size #samples x #dimension
+`obsdim = 2` means the matrix `X` (and `Y`) has size #dimension x #samples
 """
 kernelmatrix
 
@@ -109,12 +107,11 @@ kernelmatrix(κ::TransformedKernel, X::AbstractMatrix, Y::AbstractMatrix; obsdim
         kernelmatrix(kernel(κ), apply(κ.transform, X, obsdim = obsdim), apply(κ.transform, Y, obsdim = obsdim), obsdim = obsdim)
 
 """
-```
-    kerneldiagmatrix(κ::Kernel, X::Matrix; obsdim::Int=2)
-```
+    kerneldiagmatrix(κ::Kernel, X::Matrix; obsdim::Int = 2)
+
 Calculate the diagonal matrix of `X` with respect to kernel `κ`
-`obsdim=1` means the matrix `X` has size #samples x #dimension
-`obsdim=2` means the matrix `X` has size #dimension x #samples
+`obsdim = 1` means the matrix `X` has size #samples x #dimension
+`obsdim = 2` means the matrix `X` has size #dimension x #samples
 """
 function kerneldiagmatrix(
         κ::Kernel,
@@ -130,10 +127,9 @@ function kerneldiagmatrix(
 end
 
 """
-```
-    kerneldiagmatrix!(K::AbstractVector,κ::Kernel, X::Matrix; obsdim::Int=2)
-```
-In place version of `kerneldiagmatrix`
+    kerneldiagmatrix!(K::AbstractVector,κ::Kernel, X::Matrix; obsdim::Int = 2)
+
+In place version of [`kerneldiagmatrix`](@ref)
 """
 function kerneldiagmatrix!(
         K::AbstractVector,

test/kernels/custom.jl

@@ -5,7 +5,6 @@ KernelFunctions.kappa(::MyKernel, d2::Real) = exp(-d2)
 KernelFunctions.metric(::MyKernel) = SqEuclidean()
 
 # some syntactic sugar
-(κ::MyKernel)(d::Real) = kappa(κ, d)
 (κ::MyKernel)(x::AbstractVector{<:Real}, y::AbstractVector{<:Real}) = kappa(κ, x, y)
 (κ::MyKernel)(X::AbstractMatrix{<:Real}, Y::AbstractMatrix{<:Real}; obsdim = 2) = kernelmatrix(κ, X, Y; obsdim = obsdim)
 (κ::MyKernel)(X::AbstractMatrix{<:Real}; obsdim = 2) = kernelmatrix(κ, X; obsdim = obsdim)
@@ -17,7 +16,6 @@ KernelFunctions.metric(::MyKernel) = SqEuclidean()
     @test kernelmatrix(MyKernel(), [1 2; 3 4], [5 6; 7 8]) == kernelmatrix(SqExponentialKernel(), [1 2; 3 4], [5 6; 7 8])
     @test kernelmatrix(MyKernel(), [1 2; 3 4]) == kernelmatrix(SqExponentialKernel(), [1 2; 3 4])
 
-    @test MyKernel()(3) == SqExponentialKernel()(3)
     @test MyKernel()([1, 2], [3, 4]) == SqExponentialKernel()([1, 2], [3, 4])
     @test MyKernel()([1 2; 3 4], [5 6; 7 8]) == SqExponentialKernel()([1 2; 3 4], [5 6; 7 8])
     @test MyKernel()([1 2; 3 4]) == SqExponentialKernel()([1 2; 3 4])