In metrumresearchgroup/shredder: Rstan Manipulation API

library(rstan)
library(shredder)
library(tidybayes)

tidybayes

rats <- shredder::rats_example(nCores = 1)

Spread

rats%>%
  tidybayes::spread_draws(mu_alpha,mu_beta)
#> # A tibble: 4,000 x 5
#>    .chain .iteration .draw mu_alpha mu_beta
#>     <int>      <int> <int>    <dbl>   <dbl>
#>  1      1          1     1     243.    5.96
#>  2      1          2     2     243.    6.41
#>  3      1          3     3     247.    6.16
#>  4      1          4     4     244.    6.15
#>  5      1          5     5     246.    6.15
#>  6      1          6     6     241.    6.19
#>  7      1          7     7     245.    6.20
#>  8      1          8     8     239.    6.19
#>  9      1          9     9     244.    6.31
#> 10      1         10    10     240.    5.93
#> # … with 3,990 more rows

rats%>%
  stan_thin_n(20)%>%
  tidybayes::spread_draws(mu_alpha,mu_beta)
#> # A tibble: 200 x 5
#>    .chain .iteration .draw mu_alpha mu_beta
#>     <int>      <int> <int>    <dbl>   <dbl>
#>  1      1          1     1     243.    5.96
#>  2      1          2     2     242.    6.07
#>  3      1          3     3     241.    6.18
#>  4      1          4     4     245.    6.15
#>  5      1          5     5     237.    6.14
#>  6      1          6     6     239.    6.09
#>  7      1          7     7     243.    6.32
#>  8      1          8     8     240.    6.09
#>  9      1          9     9     242.    6.31
#> 10      1         10    10     241.    6.03
#> # … with 190 more rows

rats%>%
  stan_thin_n(20)%>%
  tidybayes::spread_draws(alpha[i],beta[i])
#> # A tibble: 6,000 x 6
#> # Groups:   i [30]
#>        i alpha .chain .iteration .draw  beta
#>    <int> <dbl>  <int>      <int> <int> <dbl>
#>  1     1  240.      1          1     1  6.63
#>  2     1  241.      1          2     2  5.81
#>  3     1  240.      1          3     3  6.25
#>  4     1  241.      1          4     4  6.14
#>  5     1  240.      1          5     5  5.91
#>  6     1  242.      1          6     6  5.48
#>  7     1  237.      1          7     7  6.30
#>  8     1  241.      1          8     8  5.91
#>  9     1  239.      1          9     9  6.24
#> 10     1  242.      1         10    10  6.04
#> # … with 5,990 more rows

Gather

rats%>%
  tidybayes::gather_draws(mu_alpha,mu_beta)
#> # A tibble: 8,000 x 5
#> # Groups:   .variable [2]
#>    .chain .iteration .draw .variable .value
#>     <int>      <int> <int> <chr>      <dbl>
#>  1      1          1     1 mu_alpha    243.
#>  2      1          2     2 mu_alpha    243.
#>  3      1          3     3 mu_alpha    247.
#>  4      1          4     4 mu_alpha    244.
#>  5      1          5     5 mu_alpha    246.
#>  6      1          6     6 mu_alpha    241.
#>  7      1          7     7 mu_alpha    245.
#>  8      1          8     8 mu_alpha    239.
#>  9      1          9     9 mu_alpha    244.
#> 10      1         10    10 mu_alpha    240.
#> # … with 7,990 more rows

rats%>%
  stan_thin_n(20)%>%
  tidybayes::gather_draws(mu_alpha,mu_beta)
#> # A tibble: 400 x 5
#> # Groups:   .variable [2]
#>    .chain .iteration .draw .variable .value
#>     <int>      <int> <int> <chr>      <dbl>
#>  1      1          1     1 mu_alpha    243.
#>  2      1          2     2 mu_alpha    242.
#>  3      1          3     3 mu_alpha    241.
#>  4      1          4     4 mu_alpha    245.
#>  5      1          5     5 mu_alpha    237.
#>  6      1          6     6 mu_alpha    239.
#>  7      1          7     7 mu_alpha    243.
#>  8      1          8     8 mu_alpha    240.
#>  9      1          9     9 mu_alpha    242.
#> 10      1         10    10 mu_alpha    241.
#> # … with 390 more rows

rats%>%
  stan_thin_n(20)%>%
  tidybayes::gather_draws(alpha[i],beta[i])
#> # A tibble: 12,000 x 6
#> # Groups:   i, .variable [60]
#>        i .chain .iteration .draw .variable .value
#>    <int>  <int>      <int> <int> <chr>      <dbl>
#>  1     1      1          1     1 alpha       240.
#>  2     1      1          2     2 alpha       241.
#>  3     1      1          3     3 alpha       240.
#>  4     1      1          4     4 alpha       241.
#>  5     1      1          5     5 alpha       240.
#>  6     1      1          6     6 alpha       242.
#>  7     1      1          7     7 alpha       237.
#>  8     1      1          8     8 alpha       241.
#>  9     1      1          9     9 alpha       239.
#> 10     1      1         10    10 alpha       242.
#> # … with 11,990 more rows