Как обновить объект brmsfit с измененной моделью stan, сгенерированной brms, маргинализуя распределение весов - PullRequest
0 голосов
/ 02 февраля 2019

Буду признателен за любую помощь в обновлении моего объекта brmsfit с помощью модифицированной модели stan, сгенерированной brms, потому что я хочу передать различные столбцы весов с вероятностью таким образом, что brms пока не поддерживает, вероятно.

Моя цель - получить апостериорное распределение эффектов после маргинализации по распределению весов, как обсуждено здесь .

Мне нужно сделать это в brms или stanarm, а ненапрямую, потому что я хочу использовать функции https://github.com/mjskay/tidybayes, которые в настоящее время не поддерживаются объектом stanfit.

Идея, вероятно, аналогична той, которая обсуждалась на Создание объекта brmsfit смодифицированная сгенерированная brms модель Стэна , но я не уверен, как это сделать на практике, поэтому любая помощь будет принята с благодарностью.

Кроме того, я не понимаю, как реализовать на практикевторая и третья строки этого объяснения @paul.buerkner:

Создайте «пустой» объект brmsfit вашей модели с помощью brm (..., chain = 0).Сгенерируйте и измените код Stan и замените подгонку слотов выводом rstan :: stan_model (вы также можете изменить код Stan в модели слотов).Запустите обновление (, перекомпилировать = FALSE)

# модель brms, которая будет изменена и обновлена ​​

dt = read.table(header = TRUE, text = "
n r r/n group treat c2 c1 weights
62 3 0.048387097 1 0 0.1438 1.941115288 1.941115288
96 1 0.010416667 1 0 0.237 1.186583128 1.186583128
17 0 0 0 0 0.2774 1.159882668 3.159882668
41 2 0.048780488 1 0 0.2774 1.159882668 3.159882668
212 170 0.801886792 0 0 0.2093 1.133397521 1.133397521
143 21 0.146853147 1 1 0.1206 1.128993008 1.128993008
143 0 0 1 1 0.1707 1.128993008 2.128993008
143 33 0.230769231 0 1 0.0699 1.128993008 1.128993008
73 62 1.260273973 0 1 0.1351 1.121927228 1.121927228
73 17 0.232876712 0 1 0.1206 1.121927228 1.121927228")

m <-brm(r | trials(n) + weights(weights) ~ treat*c2+(1|group), 
              data=dt, family=binomial(link=logit))

# сгенерированная модификация brmsСтэн модель modified_brms.stan

 functions { 
} 
data { 
  int<lower=1> N;  // total number of observations 
  int Y[N];  // response variable 
  int trials[N];  // number of trials 
  real<lower=0> weights[N, 10];  // model weights 
  int<lower=1> K;  // number of population-level effects 
  matrix[N, K] X;  // population-level design matrix 
  // data for group-level effects of ID 1
  int<lower=1> J_1[N];
  int<lower=1> N_1;
  int<lower=1> M_1;
  vector[N] Z_1_1;
  int prior_only;  // should the likelihood be ignored? 
} 
transformed data { 
  int Kc = K - 1; 
  matrix[N, K - 1] Xc;  // centered version of X 
  vector[K - 1] means_X;  // column means of X before centering 
  for (i in 2:K) { 
    means_X[i - 1] = mean(X[, i]); 
    Xc[, i - 1] = X[, i] - means_X[i - 1]; 
  } 
} 
parameters { 
  vector[Kc] b;  // population-level effects 
  real temp_Intercept;  // temporary intercept 
  vector<lower=0>[M_1] sd_1;  // group-level standard deviations
  vector[N_1] z_1[M_1];  // unscaled group-level effects
} 
transformed parameters { 
  // group-level effects 
  vector[N_1] r_1_1 = sd_1[1] * (z_1[1]);
} 
model { 
  vector[N] mu = temp_Intercept + Xc * b;
  for (n in 1:N) { 
    mu[n] += r_1_1[J_1[n]] * Z_1_1[n];
  } 
  // priors including all constants 
  target += student_t_lpdf(temp_Intercept | 3, 0, 10); 
  target += student_t_lpdf(sd_1 | 3, 0, 10)
    - 1 * student_t_lccdf(0 | 3, 0, 10); 
  target += normal_lpdf(z_1[1] | 0, 1);
  // likelihood including all constants 
  if (!prior_only) { 
    for (n in 1:N) 
    for (w in 1:10) {
      target += weights[n, w] * binomial_logit_lpmf(Y[n] | trials[n], mu[n]);
    }
  } 
} 
generated quantities { 
  // actual population-level intercept 
  real b_Intercept = temp_Intercept - dot_product(means_X, b); 
}

# Я могу сделать это успешно, но не знаю, как поступить отсюда:

mod_modified_brms_stan <- stan_model(‘modified_brms.stan’)

Этот вопрос также был размещен здесь .Заранее спасибо за любую помощь.

...