Turing + Makie = Turkie
Author theogf
25 Stars
Updated Last
5 Months Ago
Started In
September 2020

Docs Latest Docs Stable Coverage Status BuildStatus

Turing + Makie -> Turkie!

WIP for an inference visualization package.

To plot during sampling :

  • Trace of the chains
  • Statistics (mean and var)
  • Marginals (KDE/Histograms)
  • Autocorrelation plots
  • Show multiple chains

Additional features :

  • Selecting which variables are plotted
  • Selecting what plots to show
  • Giving a recording option
  • Additional fine tuning features like
    • Thinning
    • Creating a buffer to limit the viewing

Extra Features

  • Using a color mapping given some statistics
  • Allow to apply transformation before plotting


Small example:

using Turing
using Turkie
using Makie # You could also use CairoMakie or another backend
@model function demo(x) # Some random Turing model
    m0 ~ Normal(0, 2)
    s ~ InverseGamma(2, 3)
    m ~ Normal(m0, s)
    for i in eachindex(x)
        x[i] ~ Normal(m, s)

xs = randn(100) .+ 1 # Create some random data
m = demo(xs) # Create the model
cb = TurkieCallback(m) # Create a callback function to be given to the sample
chain = sample(m, NUTS(0.65), 300; callback = cb) # Sample and plot at the same time

If you want to show only some variables you can give a Dict to TurkieCallback :

cb = TurkieCallback(
            (m0 = [:trace, :mean], s = [:autocov, :var])

You can also directly pass OnlineStats object :

using OnlineStats
cb = TurkieCallback(
            (v = [Mean(), AutoCov(20)],)

If you want to record the video do

using Makie
record(cb.scene, joinpath(@__DIR__, "video.webm")) do io
    addIO!(cb, io)
    sample(m,  NUTS(0.65), 300; callback = cb)

Used By Packages

No packages found.