cla_nb: Naive Bayes Classifier

View source: R/cla_nb.R

cla_nbR Documentation

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

daltoolbox documentation built on May 29, 2024, 1:57 a.m.