Extracting chains individual initial step size from Turing chain object saved (serialize) after run

Hello,

I was wondering is there any way to extract individual step size for chains I run in parallel ?

I tried summarize() but it just gives mean and sd value.

Thank you

You can turn your chains object into a DataFrame to access that information in a convenient manner:

fit = sample(model, NUTS(), MCMCThreads(), 1_000, 4)
df = DataFrame(fit)
unique(df[:,["chain","step_size"]])
# 4×2 DataFrame
#  Row │ chain  step_size 
#      │ Int64  Float64   
# ─────┼──────────────────
#    1 │     1  0.242788
#    2 │     2  0.0965947
#    3 │     3  0.244368
#    4 │     4  0.0661436
1 Like