Detecting correlation between initial parameters and run time in RStan

initial_param_vs_runtime.R

library(purrr)

runtimes <- get_elapsed_time(fit)[,2]

inits <- get_inits(fit)

## traditional conversion to a matrix
# m.init <- do.call(rbind, lapply(inits, function(l) do.call(c, l)))

## using purrr
m.init <- inits %>% map(lift_dl(c)) %>% map_call(rbind)

time_cor <- cor(m.init, runtimes)
time_cor <- time_cor[ order(time_cor), ]

head(time_cor, n=25)
tail(time_cor, n=25)

# plot(m.init[,"parameter_name"], runtimes)