R/evolve_cohorts.R

In schifferl/LagSelectionBias:

evolve_cohorts <- function(simulation_data, education_covariance) {
  for (x in unique(simulation_data$date_of_birth)) {
    for (y in unique(simulation_data$race)) {
      for (z in unique(simulation_data$sex)) {
        simulation_data %>%
          filter_cohort(x, y, z) %>%
          educate_cohort(x, education_covariance) %>%
          mortify_cohort(x) %>%
          merge.data.frame(simulation_data, ., all.y = TRUE) %>%
          select(date_of_birth, race, sex, theta_education, theta_mortality, weighted_theta, education, date_of_death) %>%
          arrange(date_of_birth, race, sex, theta_mortality, weighted_theta)
      }
    }
  }
}