reg_dtree: Decision Tree for regression

View source: R/reg_dtree.R

reg_dtreeR Documentation

Decision Tree for regression

Description

Creates a regression object that uses the Decision Tree method for regression It wraps the tree library.

Usage

reg_dtree(attribute)

Arguments

attribute

attribute target to model building.

Value

A decision tree regression object

Examples

data(Boston)
model <- reg_dtree("medv")

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

model <- fit(model, train)

test_prediction <- predict(model, test)
test_predictand <- test[,"medv"]
test_eval <- evaluate(model, test_predictand, test_prediction)
test_eval$metrics

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