Set the weights for each cohort