A helper function for getting y predictions
Usage
get_y_preds(fitted_model, outcome_number, train_or_test, Z)
Arguments
- fitted_model
The fitted mvbcf model to get predictions from
- outcome_number
The outcome variable to get predictions for
- train_or_test
Indicate if predictions should be for the "train" or "test" data
- Z
A binary treatment status indicator: 1 for treatment 0 for control
Value
A vector of individual y estimates for the specified outcome
