Out of Sample Predictions on heirarchical regression model (PyMC - v5.0.1)

Do you use an index variable in the model, like mu = alpha[idx] + beta * feature_data? If so, you’ll have to make idx a mutable data object as well and pass in new indexes for your test data.