Description Usage Arguments Value Examples
View source: R/classification.R
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.
1 | decisiontree(data_train, data_test, includeplot = FALSE, showtree = FALSE)
|
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) |
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.
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)
|
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.