decisiontree: Decision tree Trains a decision on the given training dataset...

Description Usage Arguments Value Examples

View source: R/classification.R

Description

Decision tree Trains a decision on the given training dataset and uses it to predict classification for test dataset. The resulting accuracy, sensitivity and specificity are returned, as well as a tree summary.

Usage

1
decisiontree(data_train, data_test, includeplot = FALSE, showtree = FALSE)

Arguments

data_train

Training set: dataframe containing classification column and all other columns features. This is the dataset on which the decision tree model is trained.

data_test

Test set: dataframe containing classification column and all other columns features. This is the dataset on which the decision tree model in tested.

includeplot

Show performance scatter plot (default:FALSE)

showtree

Show trained decision tree graphically (default:FALSE)

Value

List containing performance summary, accessed using training (training accuracy), test (test accuracy), trainsensitivity, testsensitivity, trainspecificity, testspecificity. Also accessed using fit is the trained model produced. This can be used to find the features which appear at each level of the tree.

Examples

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
data_train = data.frame(
      classification=as.factor(c(1,1,0,0,1,1,0,0,1,1)),
      A=c(1,1,1,0,0,0,1,1,1,0),
      B=c(0,1,1,0,1,1,0,1,1,0),
      C=c(0,0,1,0,0,1,0,0,1,0))
data_test = data.frame(
      classification=as.factor(c(1,1,0,0,1,1,1,0)),
      A=c(0,0,0,1,0,0,0,1),
      B=c(1,1,1,0,0,1,1,1),
      C=c(0,0,1,1,0,0,1,1))
decisiontree(data_train,data_test)

feamiR documentation built on Jan. 19, 2021, 9:08 a.m.

Related to decisiontree in feamiR...