Implement baseline risk meta-regression for other models

ndunnewind · Oct 18, 2024 · dbab7da · dbab7da
1 parent efdcc87
commit dbab7da
Show file tree

Hide file tree

Showing 5 changed files with 21 additions and 0 deletions.
diff --git a/inst/stan/binomial_1par.stan b/inst/stan/binomial_1par.stan
@@ -34,6 +34,7 @@ transformed parameters {
       X_agd_arm * beta_tilde + offset_agd_arm :
       X_agd_arm * beta_tilde;
 
+    // Baseline risk meta-regression
     if (brmr_n_col > 0) {
       eta_agd_arm_noRE += (X_agd_arm[,1:totns] * mu - xbar_mu - 1) .* (X_agd_arm[,brmr_col] * beta_tilde[brmr_col]);
     }

diff --git a/inst/stan/binomial_2par.stan b/inst/stan/binomial_2par.stan
@@ -37,6 +37,11 @@ transformed parameters {
       X_agd_arm * beta_tilde + offset_agd_arm :
       X_agd_arm * beta_tilde;
 
+    // Baseline risk meta-regression
+    if (brmr_n_col > 0) {
+      eta_agd_arm_noRE += (X_agd_arm[,1:totns] * mu - xbar_mu - 1) .* (X_agd_arm[,brmr_col] * beta_tilde[brmr_col]);
+    }
+
     if (nint_max > 1) { // -- If integration points are used --
       if (RE) {
 

diff --git a/inst/stan/normal.stan b/inst/stan/normal.stan
@@ -39,6 +39,11 @@ transformed parameters {
       X_agd_arm * beta_tilde + offset_agd_arm :
       X_agd_arm * beta_tilde;
 
+    // Baseline risk meta-regression
+    if (brmr_n_col > 0) {
+      eta_agd_arm_noRE += (X_agd_arm[,1:totns] * mu - xbar_mu - 1) .* (X_agd_arm[,brmr_col] * beta_tilde[brmr_col]);
+    }
+
     if (nint_max > 1) { // -- If integration points are used --
       if (RE) {
 

diff --git a/inst/stan/ordered_multinomial.stan b/inst/stan/ordered_multinomial.stan
@@ -87,6 +87,11 @@ transformed parameters {
           X_agd_arm * beta_tilde + offset_agd_arm :
           X_agd_arm * beta_tilde;
 
+    // Baseline risk meta-regression
+    if (brmr_n_col > 0) {
+      eta_agd_arm_noRE += (X_agd_arm[,1:totns] * mu - xbar_mu - 1) .* (X_agd_arm[,brmr_col] * beta_tilde[brmr_col]);
+    }
+
     if (nint_max > 1) { // -- If integration points are used --
 
       if (RE) {

diff --git a/inst/stan/poisson.stan b/inst/stan/poisson.stan
@@ -34,6 +34,11 @@ transformed parameters {
       X_agd_arm * beta_tilde + offset_agd_arm :
       X_agd_arm * beta_tilde;
 
+    // Baseline risk meta-regression
+    if (brmr_n_col > 0) {
+      eta_agd_arm_noRE += (X_agd_arm[,1:totns] * mu - xbar_mu - 1) .* (X_agd_arm[,brmr_col] * beta_tilde[brmr_col]);
+    }
+
     if (nint_max > 1) { // -- If integration points are used --
       if (RE) {