Learn R Programming

iai (version 1.4.0)

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

Details

Julia Equivalent: IAI.fit_predict!

Examples

Run this code
# NOT RUN {
iai::fit_predict(lnr, X, treatments, outcomes)
# }
# NOT RUN {
# }

Run the code above in your browser using DataLab