fit_predict.numeric_reward_estimator {iai}

Fit a numeric reward estimator on features, treatments and outcomes and return predicted counterfactual rewards for each observation, under each treatment candidate, as well as the scores of the internal estimators.

Description

Julia Equivalent: IAI.fit_predict!

Usage

## S3 method for class 'numeric_reward_estimator'
fit_predict(obj, X, treatments, ...)

Arguments

obj

The learner or grid to use for estimation

X

The features of the data.

treatments

The treatment applied to each point in 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.1 or higher.

Examples

## Not run: iai::fit_predict(obj, X, treatments, outcomes)