human_annotations.stan 1.3 KB

1234567891011121314151617181920212223242526272829303132333435
  1. for (g in 1:n_rates) {
  2. real chi_mu = mu_pop_level[1]*exp(
  3. (alpha_dev+sigma_dev*child_dev_speech_age[speech_rate_child[g]])*speech_rate_age[g]/12.0/10.0 + (speech_rate_child_level[speech_rate_child[g],2]+speech_rate_child_level[speech_rate_child[g],3]-mu_pop_level[3]-mu_pop_level[4])*(beta_dev*speech_rate_age[g]/12.0/10.0+beta_direct/10.0)
  4. );
  5. speech_rate[1,g] ~ gamma(
  6. alpha_child_level[1],
  7. alpha_child_level[1]/chi_mu
  8. );
  9. speech_rate[2:,g] ~ gamma(
  10. alpha_child_level[2:],
  11. (alpha_child_level[2:]./(speech_rate_child_level[speech_rate_child[g],:]')) //'
  12. );
  13. speech_rates[g,:] ~ poisson(speech_rate[:,g]*durations[g]*1000);
  14. }
  15. for (c in 1:n_speech_rate_children) {
  16. int distrib = child_siblings[c]==0?2:1;
  17. speech_rate_child_level[c,1] ~ gamma(
  18. alpha_corpus_level[distrib,1],
  19. (alpha_corpus_level[distrib,1]/(mu_corpus_level[1,speech_rate_child_corpus[c]]*exp(
  20. speech_rate_child_siblings[c]==0?beta_sib_och:0
  21. )))
  22. );
  23. speech_rate_child_level[c,2:] ~ gamma(
  24. alpha_corpus_level[distrib,2:],
  25. (alpha_corpus_level[distrib,2:]./(mu_corpus_level[2:,speech_rate_child_corpus[c]]*exp(
  26. speech_rate_child_siblings[c]==0?beta_sib_adu/10.0:0
  27. )))
  28. );
  29. }
  30. child_dev_speech_age ~ normal(0, 1);