af: Acquisition function, selects $n_2$ units from the test set

Description Usage Arguments Value

View source: R/models.R

Description

Acquisition function, selects $n_2$ units from the test set

Usage

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
af(
  data,
  tau_train,
  tau_test,
  theta,
  n2,
  type = "emcite",
  weight_types = 5,
  B = 2
)

Arguments

data

list containing train and test covariates

tau_train

estimated ITE for train set

tau_test

predicted ITE for test set

theta

initialization of theta

n2

number of samples to select

type

which method to run: random/variance/type-s/emcite

weight_types

weight of type-s error in EMCITE

B

number of bootstrap predictions in EMCITE

Value

vector of selected indexes


Nth-iteration-labs/emcite documentation built on Sept. 28, 2020, 9:38 a.m.