make_cate {OPL}R Documentation

Function to calculate the Causal Treatment Effect

Description

Predicting conditional average treatment effect (CATE) on a new policy based on the training over an old policy

Usage

make_cate(
  model,
  train_data,
  test_data,
  w,
  x,
  y,
  family = gaussian(),
  ntree = 100,
  mtry = 2,
  verbose = TRUE
)

Arguments

model

A model object used for estimation.

train_data

The training dataset.

test_data

The test dataset.

w

Set the treatment variable.

x

set Independent variables for the model.

y

Set the outcome variable.

family

The family type for the model (e.g., 'binomial').

ntree

Number of trees for the Random Forest model.

mtry

Number of variables to consider at each tree split in the Random Forest model.

verbose

Set TRUE to print the output on the console.

Value

An object containing the estimated causal treatment effect results.

References


[Package OPL version 1.0.2 Index]