Print of the full kernel structure via recursion

Author: theogf

src/generic.jl

@@ -11,6 +11,9 @@ _scale(t::ScaleTransform, metric::Euclidean, x, y) =  first(t.s) * evaluate(metr
 _scale(t::ScaleTransform, metric::Union{SqEuclidean,DotProduct}, x, y) =  first(t.s)^2 * evaluate(metric, x, y)
 _scale(t::ScaleTransform, metric, x, y) = evaluate(metric, apply(t, x), apply(t, y))
 
+printshifted(io::IO,κ::Kernel,shift::Int) = print(io,"$κ")
+Base.show(io::IO,κ::Kernel) = print(io,nameof(typeof(κ)))
+
 ### Syntactic sugar for creating matrices and using kernel functions
 for k in subtypes(BaseKernel)
     @eval begin

src/kernels/kernelproduct.jl

@@ -49,3 +49,15 @@ function kerneldiagmatrix(
     obsdim::Int=defaultobs) #TODO Add test
     reduce(hadamard,kerneldiagmatrix(κ.kernels[i],X,obsdim=obsdim) for i in 1:length(κ))
 end
+
+function Base.show(io::IO,κ::KernelProduct)
+    printshifted(io,κ,0)
+end
+
+function printshifted(io::IO,κ::KernelProduct, shift::Int)
+    print(io,"Product of $(length(κ)) kernels:")
+    for i in 1:length(κ)
+        print(io,"\n"*("\t"^(shift+1))*"- ")
+        printshifted(io,κ.kernels[i],shift+2)
+    end
+end

src/kernels/kernelsum.jl

@@ -72,8 +72,13 @@ function kerneldiagmatrix(
 end
 
 function Base.show(io::IO,κ::KernelSum)
+    printshifted(io,κ,0)
+end
+
+function printshifted(io::IO,κ::KernelSum, shift::Int)
     print(io,"Sum of $(length(κ)) kernels:")
     for i in 1:length(κ)
-        print(io,"\n\t- (w=$(κ.weights[i])) $(κ.kernels[i])")
+        print(io,"\n"*("\t"^(shift+1))*"- (w=$(κ.weights[i])) ")
+        printshifted(io,κ.kernels[i],shift+2)
     end
 end

src/kernels/scaledkernel.jl

@@ -17,4 +17,9 @@ opt_params(k::ScaledKernel) = (k.σ,opt_params(k.kernel))
 
 Base.:*(w::Real,k::Kernel) = ScaledKernel(k,w)
 
-Base.show(io::IO,κ::ScaledKernel) = print(io,"$(κ.kernel)\n\t- σ = $(first(κ.σ))")
+Base.show(io::IO,κ::ScaledKernel) = printshifted(io,κ,0)
+
+function printshifted(io::IO,κ::ScaledKernel,shift::Int)
+    printshifted(io,κ.kernel,shift)
+    print(io,"\n"*("\t"^(shift+1))*"- σ = $(first(κ.σ))")
+end

src/kernels/transformedkernel.jl

@@ -11,4 +11,9 @@ metric(κ::TransformedKernel) = metric(κ.kernel)
 
 params(κ::TransformedKernel) = (params(κ.transform),params(κ.kernel))
 
-Base.show(io::IO,κ::TransformedKernel) = print(io,"$(κ.kernel)\n\t- $(κ.transform)")
+Base.show(io::IO,κ::TransformedKernel) = printshifted(io,κ,0)
+
+function printshifted(io::IO,κ::TransformedKernel,shift::Int)
+    printshifted(io,κ.kernel,shift)
+    print(io,"\n"*("\t"^(shift+1))*"- $(κ.transform)")
+end