macs (version 1.0)

plot.ctree: Plot function for a ctree object

Description

Draw a tree structure plot for ctree object. The plot includes main parameters of RTREE model.

Usage

# S3 method for ctree
plot(x, shape = 1, main = "ctree", ...)

Arguments

x

ctree object, typically result of tarv and rtree.

shape

has two options: 1 or 2. Determine the shape of tree where '1' uses circle and square to denote nodes while '2' uses point to represent nodes.

main

title to ctree plot.

...

Further arguments to be passed to plot.ctree

Details

This function is a method for the generic function plot, for objects of class ctree.

References

Zhang, H. and Singer, B. (1999), Recursive partitioning in the health sciences, Springer Verlag.

See Also

tarv, rtree

Examples

# NOT RUN {
# load "macs" library
library("macs")

# construct RTREE model
set.seed(1234)
data <- data.frame(y = sample(1:3, 1000, replace = TRUE),
                   n = as.factor(sample(1:4, 1000, replace = TRUE,
                                        prob = c(0.1, 0.3, 0.2, 0.4))),
                   o1 = sample(1:50, 1000, replace = TRUE),
                   o2 = sample(1:30, 1000, replace = TRUE),
                   o3 = sample(1:10, 1000, replace = TRUE),
                   o4 = sample(1:60, 1000, replace = TRUE),
                   o5 = sample(1:20, 1000, replace = TRUE),
                   o6 = sample(1:40, 1000, replace = TRUE))
head(data)
result <- rtree(data, method = "entropy", alpha = 0.01, cost = c(1, 1, 1))
result

# plot result of rtree
plot(result, main = "example of ctree plot")
plot(result, shape = 2)
# }