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
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