fit_predict: Fit a reward estimation model on features, treatments and outcomes and
return predicted counterfactual rewards for each observation, as well as the
score of the internal outcome estimator.
Description
For categorical treatments, returns the estimated reward under each
treatment observed in the data. For numeric treatments, returns the
estimated reward under each treatment candidate.
Usage
fit_predict(lnr, X, treatments, outcomes, ...)
Arguments
lnr
The learner or grid to use for estimation
X
The features of the data.
treatments
The treatment applied to each point in the data.
outcomes
The outcome observed for each point in the data.
...
For numeric treatments, the treatment candidates to consider