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.
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.
fit_predict(lnr, X, treatments, outcomes, ...)
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 |
Julia Equivalent:
IAI.fit_predict!
## Not run: iai::fit_predict(lnr, X, treatments, outcomes)
Please choose more modern alternatives, such as Google Chrome or Mozilla Firefox.