Skip to contents

A helper function for getting tau predictions

Usage

get_tau_preds(fitted_model, outcome_number, train_or_test)

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

Value

A vector of individual treatment effect estimates for the specified outcome