BinUpliftEnhanced {tools4uplift} | R Documentation |
Univariate optimal partitionning for Uplift Models. The algorithm quantizes several variables into bins and creates an augmented dataset with the binned variables.
BinUpliftEnhanced(data, treat, outcome, var.list, n.split = 10, alpha = 0.05, n.min = 30, ylim = NULL, ylab = "Uplift", title = NULL, color = NULL)
data |
a data frame containing the treatment, the outcome and the predictor to quantize. |
treat |
name of a binary (numeric) vector representing the treatment assignment (coded as 0/1). |
outcome |
name of a binary response (numeric) vector (coded as 0/1). |
var.list |
a vector of names representing the explanatory variables to quantize. |
n.split |
number of splits to test at each node. For continuous explanatory variables only (must be > 0). |
alpha |
significance level of the statistical test (must be between 0 and 1). |
n.min |
minimum number of observations per child node. |
ylim |
a range for the y axis. |
ylab |
a title for the y axis. |
title |
an overall title for the plot. |
color |
a color for the plot. If ommitted, the color will be set by default to a custom light blue. |
an augmented data frame with quantized variables. If a variable is enhanced, the function returns automatically a barplot.
Olivier Gandouet
Belbahri, M., Murua, A., Gandouet, O., and Partovi Nia, V. (2019) Uplift Regression, <https://dms.umontreal.ca/~murua/research/UpliftRegression.pdf>
BinUplift
library(tools4uplift) data("SimUplift") train.enhanced <- BinUpliftEnhanced(data = SimUplift, treat = "treat", outcome = "y", var.list = colnames(SimUplift[,3:7]))