tree2data: Prepare data from GBM or randomForest model object for...

In SixiangHu/DataMan: Data Summary, One-way Interactive Plot and Model Visualisation Analysis

Description

Prepare data for sankeyPlot from GBM or randomForest model object.

Usage

tree2data(tree_model,treeInd = 1)

## Default S3 method:
tree2data(tree_model, treeInd = 1)

## S3 method for class 'randomForest'
tree2data(tree_model, treeInd = 1)

## S3 method for class 'gbm'
tree2data(tree_model, treeInd = 1)

Arguments

tree_model	This could be either GBM or randomForest model object.
treeInd	An integer to specify the tree index from forest.

Author(s)

Sixiang Hu

Examples

## Not run: 
data(iris)
iris.mod <- gbm(Species ~ ., distribution="multinomial", 
                 data=iris, n.trees=2000, shrinkage=0.01, 
                 cv.folds=5, verbose=FALSE, n.cores=1)
tree_data <- tree2data(iris.mod,1)
library(networkD3)
sankeyNetwork(tree_data[[1]],tree_data[[2]],Source="src",Target="tar",
               Value="value",NodeID="name")

## End(Not run)

SixiangHu/DataMan documentation built on May 9, 2019, 1:48 p.m.