h2o.att: Retrieve Average Treatment Effect on the Treated

In h2o: R Interface for the 'H2O' Scalable Machine Learning Platform

Retrieve Average Treatment Effect on the Treated

Description

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

Usage

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

Arguments

object	An H2OBinomialUpliftMetrics or
train	Retrieve the training ATT value
valid	Retrieve the validation ATT 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.att(perf)

## End(Not run)

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