bartMachine (version 1.2.6)

dummify_data: Dummify Design Matrix

Description

Create a data frame with factors converted to dummies.

Usage

dummify_data(data)

Value

Returns a data frame with factors converted to dummy indicator variables.

Arguments

data

Data frame to be dummified.

Author

Adam Kapelner and Justin Bleich

Details

The column names of the dummy variables are given by the ``FactorName_LevelName'' and are augmented to the end of the design matrix. See the example below.

Examples

Run this code
#generate data
set.seed(11)
x1 = rnorm(20)
x2 = as.factor(ifelse(x1 > 0, "A", "B"))
x3 = runif(20)
X = data.frame(x1,x2,x3)
#dummify data
X_dummified = dummify_data(X)
print(X_dummified)

Run the code above in your browser using DataCamp Workspace