predict.categorical_reward_estimator {iai}

Return counterfactual rewards estimated by a categorical reward estimator for each observation in the supplied data

Description

Julia Equivalent: IAI.predict

Usage

## S3 method for class 'categorical_reward_estimator'
predict(obj, X, ...)

Arguments

obj

The learner or grid to use for estimation

X

The features of the data.

...

Additional arguments depending on the treatment and outcome types. Refer to the Julia documentation for more information.

IAI Compatibility

Requires IAI version 2.0 or higher.

Examples

## Not run: iai::predict(lnr, X, treatments, outcomes)