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

In Nth-iteration-labs/emcite: Batch Mode Active Learning for Individual Treatment Effect Estimation

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

Description

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

Usage

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 Feb. 6, 2023, 9:07 a.m.