autoFE: autoFE

In 2econsulting/rAutoFE:

Description

autoFE

Usage

autoFE(train, valid, test, x, y, savePath, verbose = FALSE)

Arguments

train	data.table object
valid	data.table object
test	data.table object
savePath	save path

Examples

library(data.table)
library(rAutoFE)
library(rAutoFS)
library(h2o)
savePath <- "c:/tmp"
dir.create(savePath)
data(churn)
churn <- as.data.table(churn)
churn[, Area.Code:=as.factor(Area.Code)]
splits <- splitFrame(dt=churn, ratio = c(0.5, 0.2), seed=1234)
train <- splits[[1]]
valid <- splits[[2]]
test  <- splits[[3]]
y = "Churn."
x = colnames(train)[colnames(train)!=y]
h2o.init()
dataset <- autoFE(train=train, valid=valid, test=test, x=x, y=y, savePath=savePath, verbose=TRUE)

2econsulting/rAutoFE documentation built on May 24, 2019, 2:49 p.m.