h2o.ate {h2o} | R Documentation |
Retrieve Average Treatment Effect
Description
Retrieves ATE from an H2OBinomialUpliftMetrics. If "train" and "valid" parameters are FALSE (default), then the training ATE is returned. If more than one parameter is set to TRUE, then a named vector of ATE values are returned, where the names are "train", "valid".
Usage
h2o.ate(object, train = FALSE, valid = FALSE)
Arguments
object |
An H2OBinomialUpliftMetrics or |
train |
Retrieve the training ATE value |
valid |
Retrieve the validation ATE value |
Examples
## Not run:
library(h2o)
h2o.init()
f <- "https://s3.amazonaws.com/h2o-public-test-data/smalldata/uplift/criteo_uplift_13k.csv"
train <- h2o.importFile(f)
train$treatment <- as.factor(train$treatment)
train$conversion <- as.factor(train$conversion)
model <- h2o.upliftRandomForest(training_frame=train, x=sprintf("f%s",seq(0:10)), y="conversion",
ntrees=10, max_depth=5, treatment_column="treatment",
auuc_type="AUTO")
perf <- h2o.performance(model, train=TRUE)
h2o.ate(perf)
## End(Not run)
[Package h2o version 3.44.0.3 Index]