cla_dtree: Decision Tree for classification

View source: R/cla_dtree.R

cla_dtreeR Documentation

Decision Tree for classification

Description

Creates a classification object that uses the Decision Tree algorithm for classification. It wraps the tree library.

Usage

cla_dtree(attribute, slevels)

Arguments

attribute

attribute target to model building.

slevels

The possible values for the target classification.

Value

A classification object that uses the Decision Tree algorithm for classification.

Examples

data(iris)
slevels <- levels(iris$Species)
model <- cla_dtree("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.