cla_nb {daltoolbox}

Naive Bayes Classifier

Description

Classification using the Naive Bayes algorithm It wraps the e1071 library.

Usage

cla_nb(attribute, slevels)

Arguments

attribute

attribute target to model building.

slevels

Possible values for the target classification.

Value

A classification object.

Examples

data(iris)
slevels <- levels(iris$Species)
model <- cla_nb("Species", slevels)

# preparing dataset for random sampling
sr <- sample_random()
sr <- train_test(sr, iris)
train <- sr$train
test <- sr$test

model <- fit(model, train)

prediction <- predict(model, test)
predictand <- adjust_class_label(test[,"Species"])
test_eval <- evaluate(model, predictand, prediction)
test_eval$metrics