I would do it this way as well. You can also add a new Variable to the model for prediction, but personally, I find it cleaner to separate the inference with the prediction.
I would do it this way as well. You can also add a new Variable to the model for prediction, but personally, I find it cleaner to separate the inference with the prediction.