predict.prunedadaStump: Predict from a 'prunedadaStump' object

In nivangio/adaStump: Fitting adaBoost model with Stumps as base learners

Description

Predict from a prunedadaStump object

Usage

## S3 method for class 'prunedadaStump'
predict(stump.model, newdata)

Arguments

stump.model	object of class prunedadaStump
newdata	Data to be fit

Value

A vector of length nrow(newdata) expressing the probability of ocurrence of the event modelled.

Examples

#Load Iris
data(iris)

#Create Variable is Iris as numerical
iris$isSetosa <- as.numeric(iris$Species == "setosa")

#Split sample in 70 train - 30 test
train.ind <- sample(nrow(iris), nrow(iris) * 0.7)

#Train model. For obvious reasons, Species variable is not included in the fit
fit <- adaStump(isSetosa ~ Sepal.Length + Sepal.Width + Petal.Length + Petal.Width, iris[train.ind,], 
                type = "discrete", iter = 10, nu = 0.05, bag.frac = 0.6)
                
#Prune Tree and predict

fit_pruned <- pruneTree(fit)
predict(fit_pruned,iris[-train.ind,])

nivangio/adaStump documentation built on May 23, 2019, 7:06 p.m.