Skip to content
Snippets Groups Projects
all.jl 788 B
Newer Older
Luca Lenz's avatar
Luca Lenz committed
abstract type AbstractRejectionChains <: AbstractMCMC.AbstractChains end
Luca Lenz's avatar
Luca Lenz committed
include("rejection.jl")
include("tables.jl")
include("stats.jl")
Luca Lenz's avatar
Luca Lenz committed

function Base.show(io::IO, c::AbstractRejectionChains)
    print(io, "Chains (×$(nchains(c))) {", eltype(c), "} $(length(c)) elements. ")
Luca Lenz's avatar
Luca Lenz committed
function Base.display(c::AbstractRejectionChains)
    println(c)
Luca Lenz's avatar
Luca Lenz committed
    if is_multilevel(c)
        ls = maximum.(levels(c))
        println(" log density levels ", ls)
    end
    data = info(c)
    for k in keys(data)
Luca Lenz's avatar
Luca Lenz committed
        if nchains(c) == 1
            println(" ", k, " : ", data[k][1])
        else
            m, s = mean(data[k]), std(data[k])
Luca Lenz's avatar
Luca Lenz committed
            m = round.(m, sigdigits=5)
            s = round.(s, sigdigits=3)
            println(" ", k, " : ", m, " ± ", s)
        end
    end
end