h2o.atc: Retrieve Average Treatment Effect on the Control

View source: R/models.R

h2o.atcR Documentation

Retrieve Average Treatment Effect on the Control

Description

Retrieves ATC from an H2OBinomialUpliftMetrics. If "train" and "valid" parameters are FALSE (default), then the training ATC is returned. If more than one parameter is set to TRUE, then a named vector of ATC values are returned, where the names are "train", "valid".

Usage

h2o.atc(object, train = FALSE, valid = FALSE)

Arguments

object

An H2OBinomialUpliftMetrics or

train

Retrieve the training ATC value

valid

Retrieve the validation ATC 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.atc(perf)

## End(Not run)

h2o documentation built on May 29, 2024, 4:26 a.m.