12345678910111213141516171819202122232425262728293031 |
- real recs_priors_lpmf(array[] int children,
- int start, int end,
- int n_recs,
- int n_classes,
- real recs_duration,
- array [] real age,
- matrix truth_vocs,
- vector mu_pop_level,
- matrix mu_child_level,
- vector alpha_child_level,
- vector child_dev_age,
- real beta_dev,
- real beta_direct
- ) {
- real ll = 0;
-
- for (k in start:end) {
- real chi_mu = mu_pop_level[1]*exp(
- (child_dev_age[children[k-start+1]])*age[k]/12.0/10.0+(mu_child_level[children[k-start+1],2]+mu_child_level[children[k-start+1],3]-mu_pop_level[3]-mu_pop_level[4])*(beta_dev*age[k]/12.0/10.0+beta_direct/10.0)
- );
- ll += gamma_lpdf(
- truth_vocs[k,1]/1000/recs_duration | alpha_child_level[1], alpha_child_level[1]/chi_mu
- );
- ll += gamma_lpdf(
- truth_vocs[k,2:]/1000/recs_duration | alpha_child_level[2:], alpha_child_level[2:]./mu_child_level[children[k-start+1],:]' //'
- );
- }
- return ll;
- }
|