predict_reward function

Generic function for returning the counterfactual rewards estimated by a model under each treatment