Man pages for Nth-iteration-labs/emcite
Batch Mode Active Learning for Individual Treatment Effect Estimation

afAcquisition function, selects $n_2$ units from the test set
dgpCreate data according to simulation setting
fit_gradient_descentFit stochastic gradient descent
predict_itePredict ITE from trained BART model on given data
retrain_and_metricsRetrain and evaluate the model based on PEHE
sampling_dataCreate n_1+n_2 dataset
sgd_pyRun model in python
split_dataSplit the data into train and test set
train_modelTrain BART model for ITE estimation
Nth-iteration-labs/emcite documentation built on Feb. 6, 2023, 9:07 a.m.