classify: Fit GBDT model and predict classification

In jiho/soundclass: Signal classifier for sounds

Description

Fit GBDT model and predict classification

Usage

classify(data, train, verbose = TRUE, ...)

Arguments

data	a data.frame with signal characteristics of signals to be predicted
train	a data.frame with signal characteristics and classification of signals from which to learn the model
verbose	when TRUE (the default), print information about the fitted model
...	passed to fit.gbdt

Examples

data(sirena)
sub <- subsample(sirena[,-ncol(sirena)], p=0.2)
train <- sirena[sub$picked,]
data <- sirena[!sub$picked,]
pred <- classify(data=data[,-ncol(data)], train=train,
                 n.trees=100, shrinkage=0.01, n.minobsinnode=1,
                 n.cores=1)
# NB: n.cores = 1 is necessary for examples to run on all machines. Feel
#     free to remove it and use more cores on your machine.
head(pred)
(cm <- confusion_matrix(true=data$type, pred=pred$type))
confusion_stats(cm)

jiho/soundclass documentation built on June 5, 2019, 10:10 p.m.