Predicting for Schools Outside the Training Range using NumPyro's Predictive Class

numpyro
2

i think you either have to

  • add dummy schools during inference (e.g. by giving them fake data and masking our their likelihood) or;
  • use posterior samples directly to compute linear_combination for the school of interest
    the first case would still use Predictive