rbooster
This package aims to enable the use of classifiers in boosting family algorithms, apart from decision trees and other classical weak classifiers. The current scope of the package includes Real AdaBoost and Discrete AdaBoost, as well as multiclass versions of these methods, the SAMME and SAMME.R algorithms.
I am aware that there have been attempts to create frameworks that allow using custom functions in boosting (e.g. boostr). However, they are too complex to my taste. I haven’t seen a simple-to-use package that handles different boosting methods. What can be done simply in Python, why can’t it be done in R? That is my point. The package is still in the starting phase (a baby package). Things on my mind:
booster
functionThere is a function for each method in the package. But the main
function is the booster
. By using this function it will be
possible to access everything. Let me give you information about the
features of this function and how to use it. First of all, 2 different
AdaBoost methods can be applied with the booster
function:
Real AdaBoost and Discrete AdaBoost. Let us use glass0 dataset in
imbalance
package and prepare a train and test dataset.
test dataset is not mandatory.
library(rbooster)
cv_sampler <- function(y, train_proportion) {
unlist(lapply(unique(y), function(m) sample(which(y==m), round(sum(y==m))*train_proportion)))
}
library(imbalance)
data <- glass0
p <- ncol(data) - 1
x <- data[,1:p]
y <- data[, p + 1]
train_i <- cv_sampler(y, 0.9)
x_train <- x[train_i,]
y_train <- y[train_i]
x_test <- x[-train_i,]
y_test <- y[-train_i]
This is a two-class dataset. Let us use decision tree to build Discrete and Real AdaBoost models.
m_discrete <- booster(x_train = x_train,
y_train = y_train,
classifier = "rpart",
method = "discrete",
x_test = x_test,
y_test = y_test,
weighted_bootstrap = FALSE,
max_iter = 20,
lambda = 1,
print_detail = TRUE,
print_plot = TRUE,
bag_frac = 0.8,
p_weak = 4)
#> 1 Train err:0.07292, Test err:0.40909, Weighted err:0.07292
#> 2 Train err:0.07292, Test err:0.40909, Weighted err:0.11656
#> 3 Train err:0.00521, Test err:0.31818, Weighted err:0.03643
#> 4 Train err:0.00000, Test err:0.13636, Weighted err:0.07381
#> 5 Train err:0.00000, Test err:0.13636, Weighted err:0.14787
#> 6 Train err:0.00000, Test err:0.09091, Weighted err:0.05108
#> 7 Train err:0.00000, Test err:0.09091, Weighted err:0.13798
#> 8 Train err:0.00000, Test err:0.04545, Weighted err:0.10459
#> 9 Train err:0.00000, Test err:0.04545, Weighted err:0.14130
#> 10 Train err:0.00000, Test err:0.04545, Weighted err:0.05207
#> 11 Train err:0.00000, Test err:0.04545, Weighted err:0.07510
#> 12 Train err:0.00000, Test err:0.04545, Weighted err:0.18407
#> 13 Train err:0.00000, Test err:0.04545, Weighted err:0.18797
#> 14 Train err:0.00000, Test err:0.04545, Weighted err:0.19599
#> 15 Train err:0.00000, Test err:0.04545, Weighted err:0.05585
#> 16 Train err:0.00000, Test err:0.04545, Weighted err:0.09995
#> 17 Train err:0.00000, Test err:0.04545, Weighted err:0.05907
#> 18 Train err:0.00000, Test err:0.04545, Weighted err:0.21359
#> 19 Train err:0.00000, Test err:0.04545, Weighted err:0.13045
#> 20 Train err:0.00000, Test err:0.04545, Weighted err:0.09187
m_real <- booster(x_train = x_train,
y_train = y_train,
classifier = "rpart",
method = "real",
x_test = x_test,
y_test = y_test,
weighted_bootstrap = FALSE,
max_iter = 20,
lambda = 1,
print_detail = TRUE,
print_plot = TRUE,
bag_frac = 0.8,
p_weak = 4)
#> 1 Train err:0.05208, Test err:0.22727, Weighted err:0.05208
#> 2 Train err:0.05208, Test err:0.22727, Weighted err:0.30341
#> 3 Train err:0.04688, Test err:0.22727, Weighted err:0.00763
#> 4 Train err:0.03646, Test err:0.22727, Weighted err:0.01811
#> 5 Train err:0.01562, Test err:0.22727, Weighted err:0.02664
#> 6 Train err:0.01562, Test err:0.22727, Weighted err:0.05475
#> 7 Train err:0.00000, Test err:0.18182, Weighted err:0.01584
#> 8 Train err:0.00000, Test err:0.13636, Weighted err:0.06246
#> 9 Train err:0.00000, Test err:0.13636, Weighted err:0.33138
#> 10 Train err:0.00521, Test err:0.18182, Weighted err:0.07310
#> 11 Train err:0.00000, Test err:0.22727, Weighted err:0.01673
#> 12 Train err:0.00000, Test err:0.22727, Weighted err:0.06225
#> 13 Train err:0.00000, Test err:0.18182, Weighted err:0.04139
#> 14 Train err:0.00000, Test err:0.22727, Weighted err:0.08998
#> 15 Train err:0.00000, Test err:0.18182, Weighted err:0.09007
#> 16 Train err:0.00000, Test err:0.09091, Weighted err:0.02046
#> 17 Train err:0.00000, Test err:0.04545, Weighted err:0.13137
#> 18 Train err:0.00000, Test err:0.09091, Weighted err:0.01812
#> 19 Train err:0.00000, Test err:0.04545, Weighted err:0.38967
#> 20 Train err:0.00000, Test err:0.13636, Weighted err:0.01421
We can obtain predictions on x_test
like:
head(m_discrete$test_prediction)
#> [1] "positive" "positive" "positive" "positive" "positive" "positive"
head(m_real$test_prediction)
#> [1] "positive" "positive" "positive" "negative" "positive" "positive"
table(y_test, m_discrete$test_prediction)
#>
#> y_test negative positive
#> negative 14 1
#> positive 0 7
table(y_test, m_real$test_prediction)
#>
#> y_test negative positive
#> negative 13 2
#> positive 1 6
However x_test
is used to validate on
y_test
. So without y_test
, x_test
is ignored. x_test
and y_test
can be set to
NULL
, they are not mandatory for model to be established.
You can use the usual predict
function.
pred_discrete <- predict(object = m_discrete, newdata = x_test, type = "pred")
pred_real <- predict(object = m_real, newdata = x_test, type = "pred")
all(pred_discrete == m_discrete$test_prediction)
#> [1] TRUE
all(pred_discrete == m_discrete$test_prediction)
#> [1] TRUE
In predict
function type
can be “pred” or
“prob”. “prob” will gives class probability matrix.
prob_discrete <- predict(object = m_discrete, newdata = x_test, type = "prob")
head(prob_discrete)
#> negative positive
#> [1,] 3.347931e-04 0.9996652
#> [2,] 9.063150e-08 0.9999999
#> [3,] 2.162515e-06 0.9999978
#> [4,] 3.697072e-04 0.9996303
#> [5,] 4.756480e-07 0.9999995
#> [6,] 1.791538e-07 0.9999998
Let us use Glass dataset in mlbench for multiclass classification example.
library(mlbench)
data(Glass)
data <- Glass
p <- ncol(data) - 1
x <- data[,1:p]
y <- data[, p + 1]
train_i <- cv_sampler(y, 0.9)
x_train <- x[train_i,]
y_train <- y[train_i]
x_test <- x[-train_i,]
y_test <- y[-train_i]
par(mfrow = c(2,1))
m_discrete <- booster(x_train = x_train,
y_train = y_train,
classifier = "rpart",
method = "discrete",
x_test = x_test,
y_test = y_test,
weighted_bootstrap = FALSE,
max_iter = 20,
lambda = 1,
print_detail = FALSE,
print_plot = TRUE,
bag_frac = 0.8,
p_weak = p)
m_real <- booster(x_train = x_train,
y_train = y_train,
classifier = "rpart",
method = "real",
x_test = x_test,
y_test = y_test,
weighted_bootstrap = FALSE,
max_iter = 20,
lambda = 0.1,
print_detail = FALSE,
print_plot = TRUE,
bag_frac = 1,
p_weak = p)
invisible(dev.off())
pred_discrete <- predict(object = m_discrete, newdata = x_test, type = "pred")
pred_real <- predict(object = m_real, newdata = x_test, type = "pred")
table(y_test, pred_discrete)
#> pred_discrete
#> y_test 1 2 3 5 6 7
#> 1 5 1 1 0 0 0
#> 2 1 6 0 1 0 0
#> 3 1 0 1 0 0 0
#> 5 0 1 0 1 0 0
#> 6 0 0 0 0 1 0
#> 7 0 0 0 0 0 3
table(y_test, pred_real)
#> pred_real
#> y_test 1 2 3 5 6 7
#> 1 3 2 1 0 0 1
#> 2 1 5 1 1 0 0
#> 3 1 0 1 0 0 0
#> 5 0 0 0 2 0 0
#> 6 0 0 0 0 1 0
#> 7 0 0 0 0 0 3
There is a major reason people use decision trees in boosting. It
fits very quick. Let us try a pre-ready classifier Discrete Naive Bayes,
“dnb”, in rbooster
.
par(mfrow = c(2,1))
m_discrete <- booster(x_train = x_train,
y_train = y_train,
classifier = "dnb",
method = "discrete",
x_test = x_test,
y_test = y_test,
weighted_bootstrap = FALSE,
max_iter = 250,
lambda = 1,
print_detail = FALSE,
print_plot = TRUE,
bag_frac = 0.5,
p_weak = 4)
m_real <- booster(x_train = x_train,
y_train = y_train,
classifier = "dnb",
method = "real",
x_test = x_test,
y_test = y_test,
weighted_bootstrap = FALSE,
max_iter = 250,
lambda = 1e-4,
print_detail = FALSE,
print_plot = TRUE,
bag_frac = 0.2,
p_weak = 4)
invisible(dev.off())
pred_discrete <- predict(object = m_discrete, newdata = x_test, type = "pred")
pred_real <- predict(object = m_real, newdata = x_test, type = "pred")
table(y_test, pred_discrete)
#> pred_discrete
#> y_test 1 2 3 5 6 7
#> 1 2 3 2 0 0 0
#> 2 2 5 0 1 0 0
#> 3 0 1 1 0 0 0
#> 5 0 2 0 0 0 0
#> 6 0 1 0 0 0 0
#> 7 0 0 0 0 0 3
table(y_test, pred_real)
#> pred_real
#> y_test 1 2 3 5 6 7
#> 1 6 1 0 0 0 0
#> 2 3 4 0 1 0 0
#> 3 1 1 0 0 0 0
#> 5 0 2 0 0 0 0
#> 6 0 0 0 0 1 0
#> 7 0 0 0 0 0 3
I needed to play with lambda, bag_frac and p_weak a lot to find a good setting so that “dnb” fits. It is still not quite successful. Methods other that decision trees require care.
Classifier function requires x_train, y_train, weights as mandatory. Output of Classifier function will be the input for Predictor function. Predictor function requires model, which is the output of Classifier, x_new, which is the train features, and type inputs. type can only be “pred” or “prob”. “pred” must give a class vector of predictions, and “prob” must give a matrix of class probabilities. “pred” is mandatory for both Real and Discrete AdaBoost. However, Discrete AdaBoost does not require type “prob”. Let me prepare a made up Classifier function.
classifier_lm <- function(x_train, y_train, weights, ...){
y_train_code <- c(-1,1)
y_train_coded <- sapply(levels(y_train), function(m) y_train_code[(y_train == m) + 1])
y_train_coded <- y_train_coded[,1]
if (is.null(weights)) {
weights <- rep(1, length(y_train))
}
model <- lm.wfit(x = as.matrix(cbind(1,x_train)), y = y_train_coded, w = weights)
return(list(coefficients = model$coefficients,
levels = levels(y_train)))
}
predictor_lm <- function(model, x_new, type = "pred", ...) {
coef <- model$coefficients
levels <- model$levels
fit <- as.matrix(cbind(1, x_new))%*%coef
probs <- 1/(1 + exp(-fit))
probs <- data.frame(probs, 1 - probs)
colnames(probs) <- levels
if (type == "pred") {
preds <- factor(levels[apply(probs, 1, which.max)], levels = levels, labels = levels)
return(preds)
}
if (type == "prob") {
return(probs)
}
}
This classifier will convert y_train into (-1,1) codes and predictions will be transformed into the range (0,1) using logit transformation. These values will be used as probabilities and class with the higher probabilitiy will be selected as prediction. Since this is a linear classification method, it should work well with boosting. Let us see.
data <- glass0
p <- ncol(data) - 1
x <- data[,1:p]
y <- data[, p + 1]
train_i <- cv_sampler(y, 0.9)
x_train <- x[train_i,]
y_train <- y[train_i]
x_test <- x[-train_i,]
y_test <- y[-train_i]
par(mfrow = c(2,1))
m_discrete <- booster(x_train = x_train,
y_train = y_train,
classifier = classifier_lm,
predictor = predictor_lm,
method = "discrete",
x_test = x_test,
y_test = y_test,
weighted_bootstrap = FALSE,
max_iter = 600,
lambda = 2,
print_detail = FALSE,
print_plot = TRUE,
bag_frac = 0.4,
p_weak = 4)
m_real <- booster(x_train = x_train,
y_train = y_train,
classifier = classifier_lm,
predictor = predictor_lm,
method = "real",
x_test = x_test,
y_test = y_test,
weighted_bootstrap = FALSE,
max_iter = 200,
lambda = 0.1,
print_detail = FALSE,
print_plot = TRUE,
bag_frac = 1,
p_weak = 4)
#> Warning in bb(x_train = x_train, y_train = y_train, classifier = classifier, :
#> Learning stopped at iteration 134
invisible(dev.off())
pred_discrete <- predict(object = m_discrete, newdata = x_test, type = "pred")
pred_real <- predict(object = m_real, newdata = x_test, type = "pred")
table(y_test, pred_discrete)
#> pred_discrete
#> y_test negative positive
#> negative 14 1
#> positive 0 7
table(y_test, pred_real)
#> pred_real
#> y_test negative positive
#> negative 14 1
#> positive 4 3
It fits but not quite there. print_plot
can be set to
FALSE
. In that case, the same plot can be generated using
plot
function.
Pre-ready weak classifiers are “rpart”, “gnb”, “dnb”, “earth” and “glm” which are CART, Gaussian Naive Bayes, Discrete Naive Bayes, Multivariate Adaptive Regression Splines from earth package and logistic regression respectively. “glm” is not capable of multiclass classification. booster object can be used with prediction, plot, and print functions. For other details about booster object and its outputs, please see help document.