martTrain {mseq} | R Documentation |
Train the MART model.
martTrain(data, shrinkage = 0.06, interaction.depth = 10, n.trees = 2000, small_count = 0.5)
data |
the data frame generated by expData.R |
shrinkage |
the shrinkage argument of gbm |
interaction.depth |
the interaction.depth argument of gbm |
n.trees |
the n.trees argument of gbm |
small_count |
the small count which will replace the zero count |
a gbm object
# read and expand the data data(g1_part) # for real data, please use read.csv, like g1 <- read.csv("g1.csv") data <- expData(g1_part, 2, 3) data.mart <- martTrain(data, interaction.depth = 2, n.trees = 100)