expected_obs.stan

Go to the documentation of this file.

 
vector expected_obs(
  real tar_obs, vector lh, int l, int ref_as_p, int rep_agg_p,
  array[] int n_selected, array[,] int selected_idx
) {
  vector[l] p;
  if (ref_as_p == 1) {
    p = lh;
  }else{
    profile("model_likelihood_expected_obs_inv_logit") {
    p = inv_logit(lh);
    }
    profile("model_likelihood_expected_obs_hazard_to_prob") {
    p = hazard_to_log_prob(p, l);
    }
  }
  if (rep_agg_p == 1) {
    vector[l] p_aggregated = rep_vector(negative_infinity(), l);
    for (i in 1:l) {
      if (n_selected[i] > 0) {
        p_aggregated[i] = log_sum_exp(p[selected_idx[i, 1:n_selected[i]]]);
      }
    }
    p = p_aggregated;
  }
  return(tar_obs + p);
}