\documentclass[nojss]{jss}
%\VignetteIndexEntry{ctree: Conditional Inference Trees}
%\VignetteDepends{coin, TH.data, survival, strucchange, Formula, sandwich, datasets}
%\VignetteKeywords{conditional inference, non-parametric models, recursive partitioning}
%\VignettePackage{partykit}
%% packages
\usepackage{amstext}
\usepackage{amsfonts}
\usepackage{amsmath}
\usepackage{thumbpdf}
\usepackage{rotating}
%% need no \usepackage{Sweave}
\usepackage[utf8]{inputenc}
%% commands
\newcommand{\fixme}[1]{\emph{\marginpar{FIXME} (#1)}}
\renewcommand{\Prob}{\mathbb{P} }
\renewcommand{\E}{\mathbb{E}}
\newcommand{\V}{\mathbb{V}}
\newcommand{\Var}{\mathbb{V}}
\newcommand{\R}{\mathbb{R} }
\newcommand{\N}{\mathbb{N} }
%%\newcommand{\C}{\mathbb{C} }
\newcommand{\argmin}{\operatorname{argmin}\displaylimits}
\newcommand{\argmax}{\operatorname{argmax}\displaylimits}
\newcommand{\LS}{\mathcal{L}_n}
\newcommand{\TS}{\mathcal{T}_n}
\newcommand{\LSc}{\mathcal{L}_{\text{comb},n}}
\newcommand{\LSbc}{\mathcal{L}^*_{\text{comb},n}}
\newcommand{\F}{\mathcal{F}}
\newcommand{\A}{\mathcal{A}}
\newcommand{\yn}{y_{\text{new}}}
\newcommand{\z}{\mathbf{z}}
\newcommand{\X}{\mathbf{X}}
\newcommand{\Y}{\mathbf{Y}}
\newcommand{\sX}{\mathcal{X}}
\newcommand{\sY}{\mathcal{Y}}
\newcommand{\T}{\mathbf{T}}
\newcommand{\x}{\mathbf{x}}
\renewcommand{\a}{\mathbf{a}}
\newcommand{\xn}{\mathbf{x}_{\text{new}}}
\newcommand{\y}{\mathbf{y}}
\newcommand{\w}{\mathbf{w}}
\newcommand{\ws}{\mathbf{w}_\cdot}
\renewcommand{\t}{\mathbf{t}}
\newcommand{\M}{\mathbf{M}}
\renewcommand{\vec}{\text{vec}}
\newcommand{\B}{\mathbf{B}}
\newcommand{\K}{\mathbf{K}}
\newcommand{\W}{\mathbf{W}}
\newcommand{\D}{\mathbf{D}}
\newcommand{\I}{\mathbf{I}}
\newcommand{\bS}{\mathbf{S}}
\newcommand{\cellx}{\pi_n[\x]}
\newcommand{\partn}{\pi_n(\mathcal{L}_n)}
\newcommand{\err}{\text{Err}}
\newcommand{\ea}{\widehat{\text{Err}}^{(a)}}
\newcommand{\ecv}{\widehat{\text{Err}}^{(cv1)}}
\newcommand{\ecvten}{\widehat{\text{Err}}^{(cv10)}}
\newcommand{\eone}{\widehat{\text{Err}}^{(1)}}
\newcommand{\eplus}{\widehat{\text{Err}}^{(.632+)}}
\newcommand{\eoob}{\widehat{\text{Err}}^{(oob)}}
\newcommand{\bft}{\mathbf{t}}
\hyphenation{Qua-dra-tic}
\title{\texttt{ctree}: Conditional Inference Trees}
\Plaintitle{ctree: Conditional Inference Trees}
\author{Torsten Hothorn\\Universit\"at Z\"urich
\And Kurt Hornik\\Wirtschaftsuniversit\"at Wien
\And Achim Zeileis\\Universit\"at Innsbruck}
\Plainauthor{Torsten Hothorn, Kurt Hornik, Achim Zeileis}
\Abstract{
This vignette describes the new reimplementation of conditional inference trees (CTree)
in the \proglang{R} package \pkg{partykit}. CTree is a non-parametric class of regression
trees embedding tree-structured regression models into a well defined theory of
conditional inference procedures. It is applicable to all kinds of regression problems,
including nominal, ordinal, numeric, censored as well as multivariate response
variables and arbitrary measurement scales of the covariates.
The vignette comprises a practical guide to exploiting the flexible
and extensible computational tools in \pkg{partykit} for fitting and
visualizing conditional inference trees.
}
\Keywords{conditional inference, non-parametric models, recursive partitioning}
\Address{
Torsten Hothorn\\
Institut f\"ur Epidemiologie, Biostatistik und Pr\"avention \\
Universit\"at Z\"urich \\
Hirschengraben 84\\
CH-8001 Z\"urich, Switzerland \\
E-mail: \email{Torsten.Hothorn@R-project.org}\\
URL: \url{http://user.math.uzh.ch/hothorn/}\\
Kurt Hornik \\
Institute for Statistics and Mathematics\\
WU Wirtschaftsuniversit\"at Wien\\
Welthandelsplatz 1 \\
1020 Wien, Austria\\
E-mail: \email{Kurt.Hornik@R-project.org} \\
URL: \url{http://statmath.wu.ac.at/~hornik/}\\
Achim Zeileis \\
Department of Statistics \\
Faculty of Economics and Statistics \\
Universit\"at Innsbruck \\
Universit\"atsstr.~15 \\
6020 Innsbruck, Austria \\
E-mail: \email{Achim.Zeileis@R-project.org} \\
URL: \url{http://eeecon.uibk.ac.at/~zeileis/}
}
\begin{document}
<>=
suppressWarnings(RNGversion("3.5.2"))
options(width = 70, SweaveHooks = list(leftpar =
function() par(mai = par("mai") * c(1, 1.1, 1, 1))))
require("partykit")
require("coin")
require("strucchange")
require("coin")
require("Formula")
require("survival")
require("sandwich")
set.seed(290875)
@
\setkeys{Gin}{width=\textwidth}
\section{Overview}
This vignette describes conditional inference trees
\citep{Hothorn+Hornik+Zeileis:2006} along with its new and improved
reimplementation in package \pkg{partykit}. Originally, the method was
implemented in the package \pkg{party} almost entirely in \proglang{C} while
the new implementation is now almost entirely in \proglang{R}. In
particular, this has the advantage that all the generic infrastructure from
\pkg{partykit} can be reused, making many computations more modular and
easily extensible. Hence, \code{partykit::ctree} is the new reference
implementation that will be improved and developed further in the future.
In almost all cases, the two implementations will produce identical trees.
In exceptional cases, additional parameters have to be specified in order to
ensure backward compatibility. These and novel features in
\code{ctree:partykit} are introduced in Section~\ref{sec:novel}.
\section{Introduction}
The majority of recursive partitioning algorithms
are special cases of a simple two-stage algorithm:
First partition the observations by univariate splits in a recursive way and
second fit a constant model in each cell of the resulting partition.
The most popular implementations of such algorithms are `CART'
\citep{Breiman+Friedman+Olshen:1984} and `C4.5' \citep{Quinlan:1993}. Not unlike AID,
both perform an exhaustive search over all possible splits maximizing an
information measure of node impurity selecting the
covariate showing the best split.
This approach has two fundamental problems: overfitting and a selection
bias towards covariates with many possible splits.
With respect to the
overfitting problem \cite{Mingers:1987} notes that the algorithm
\begin{quote}
[\ldots] has no concept of statistical significance, and so cannot
distinguish between a significant and an insignificant improvement in the
information measure.
\end{quote}
With conditional inference trees \citep[see][for a full description of its
methodological foundations]{Hothorn+Hornik+Zeileis:2006}
we enter at the point where \cite{White+Liu:1994} demand for
\begin{quote}
[\ldots] a \textit{statistical} approach [to recursive partitioning] which
takes
into account the \textit{distributional} properties of the measures.
\end{quote}
We present a unified framework embedding recursive binary partitioning into
the well defined theory of permutation tests developed by
\cite{Strasser+Weber:1999}.
The conditional distribution of statistics measuring the association between
responses and covariates is the basis for an unbiased selection among
covariates measured at different scales.
Moreover, multiple test procedures are applied to determine whether no
significant
association between any of the covariates and the response can be stated and
the recursion needs to stop.
\section{Recursive binary partitioning} \label{algo}
We focus on regression models describing the conditional distribution of a
response variable $\Y$ given the status of $m$ covariates by
means of tree-structured recursive partitioning. The response $\Y$ from some
sample space $\sY$ may be multivariate as well.
The $m$-dimensional covariate vector $\X = (X_1, \dots, X_m)$ is taken
from a sample space $\sX = \sX_1 \times \cdots \times \sX_m$.
Both response variable and covariates may be measured
at arbitrary scales.
We assume that the conditional distribution $D(\Y | \X)$ of the response
$\Y$ given the covariates $\X$ depends on a function $f$ of the covariates
\begin{eqnarray*}
D(\Y | \X) = D(\Y | X_1, \dots, X_m) = D(\Y | f( X_1, \dots,
X_m)),
\end{eqnarray*}
where we restrict ourselves to partition based regression relationships,
i.e., $r$ disjoint cells $B_1, \dots, B_r$ partitioning the covariate space $\sX
= \bigcup_{k = 1}^r B_k$.
A model of the regression relationship is to be fitted based on a learning
sample $\LS$, i.e., a random sample of $n$ independent and
identically distributed observations, possibly with some covariates $X_{ji}$
missing,
\begin{eqnarray*}
\LS & = & \{ (\Y_i, X_{1i}, \dots, X_{mi}); i = 1, \dots, n \}.
\end{eqnarray*}
A generic algorithm for recursive binary partitioning for a given learning
sample $\LS$ can be formulated using non-negative integer valued case weights $\w
= (w_1, \dots, w_n)$. Each node of a tree is represented by a vector of
case weights having non-zero elements when the corresponding observations
are elements of the node and are zero otherwise. The following algorithm
implements recursive binary partitioning:
\begin{enumerate}
\item For case weights $\w$ test the global null hypothesis of independence between
any of the $m$ covariates and the response. Stop if this
hypothesis cannot be rejected.
Otherwise select the covariate $X_{j^*}$ with strongest
association to $\Y$.
\item Choose a set $A^* \subset \sX_{j^*}$ in order to split $\sX_{j^*}$ into
two disjoint sets $A^*$ and $\sX_{j^*} \setminus A^*$.
The case weights $\w_\text{left}$ and $\w_\text{right}$ determine the
two subgroups with $w_{\text{left},i} = w_i I(X_{j^*i} \in A^*)$ and
$w_{\text{right},i} = w_i I(X_{j^*i} \not\in A^*)$ for all $i = 1,
\dots, n$ ($I(\cdot)$ denotes the indicator function).
\item Recursively repeat steps 1 and 2 with modified case weights
$\w_\text{left}$ and $\w_\text{right}$, respectively.
\end{enumerate}
The separation of variable
selection and splitting procedure into steps 1 and 2 of the algorithm
is the key for the construction of interpretable tree
structures not suffering a systematic tendency towards covariates with many
possible splits or many missing values. In addition, a statistically
motivated and intuitive stopping criterion can be implemented: We stop
when the global null hypothesis of independence between the
response and any of the $m$ covariates cannot be rejected at a pre-specified
nominal level~$\alpha$. The algorithm induces a partition $\{B_1, \dots, B_r\}$ of
the covariate space $\sX$, where each cell $B \in \{B_1, \dots, B_r\}$
is associated with a vector of case weights.
\section{Recursive partitioning by conditional inference} \label{framework}
In the main part of this section we focus on step 1 of the generic algorithm.
Unified tests for independence are constructed by means of the conditional
distribution of linear statistics in the permutation test framework
developed by \cite{Strasser+Weber:1999}. The determination of the best binary split
in one selected covariate and the handling of missing values
is performed based on standardized linear statistics within the same
framework as well.
\subsection{Variable selection and stopping criteria}
At step 1 of the generic algorithm given in Section~\ref{algo} we face an
independence problem. We need to decide whether there is any information
about the response variable covered by any of the $m$ covariates. In each node
identified by case weights $\w$, the
global hypothesis of independence is formulated in terms of the $m$ partial hypotheses
$H_0^j: D(\Y | X_j) = D(\Y)$ with global null hypothesis $H_0 = \bigcap_{j = 1}^m
H_0^j$.
When we are not able to reject $H_0$ at a pre-specified
level $\alpha$, we stop the recursion.
If the global hypothesis can be rejected, we measure the association
between $\Y$ and each of the covariates $X_j, j = 1, \dots, m$, by
test statistics or $P$-values indicating the deviation from the partial
hypotheses $H_0^j$.
For notational convenience and without loss of generality we assume that the
case weights $w_i$ are either zero or one. The symmetric group of all
permutations of the elements of $(1, \dots, n)$ with corresponding case
weights $w_i = 1$ is denoted by $S(\LS, \w)$. A more general notation is
given in the Appendix. We measure the association between $\Y$ and $X_j, j = 1, \dots, m$,
by linear statistics of the form
\begin{eqnarray} \label{linstat}
\T_j(\LS, \w) = \vec \left( \sum_{i=1}^n w_i g_j(X_{ji})
h(\Y_i, (\Y_1, \dots, \Y_n))^\top \right) \in \R^{p_jq}
\end{eqnarray}
where $g_j: \sX_j \rightarrow \R^{p_j}$ is a non-random transformation of
the covariate $X_j$. The transformation may be specified using the
\code{xtrafo} argument (Note: this argument is currently not implemented
in \code{partykit::ctree} but is available from \code{party::ctree}).
%%If, for example, a ranking \textit{both}
%%\code{x1} and \code{x2} is required,
%%<>=
%%party:::ctree(y ~ x1 + x2, data = ls, xtrafo = function(data) trafo(data,
%%numeric_trafo = rank))
%%@
%%can be used.
The \emph{influence function}
$h: \sY \times \sY^n \rightarrow
\R^q$ depends on the responses $(\Y_1, \dots, \Y_n)$ in a permutation
symmetric way.
%%, i.e., $h(\Y_i, (\Y_1, \dots, \Y_n)) = h(\Y_i, (\Y_{\sigma(1)}, \dots,
%%\Y_{\sigma(n)}))$ for all permutations $\sigma \in S(\LS, \w)$.
Section~\ref{examples} explains how to choose $g_j$ and $h$ in different
practical settings. A $p_j \times q$ matrix is converted into a
$p_jq$ column vector by column-wise combination using the `vec' operator.
The influence function can be specified using the \code{ytrafo} argument.
The distribution of $\T_j(\LS, \w)$ under $H_0^j$ depends on the joint
distribution of $\Y$ and $X_j$, which is unknown under almost all practical
circumstances. At least under the null hypothesis one can dispose of this
dependency by fixing the covariates and conditioning on all possible
permutations of the responses. This principle leads to test procedures known
as \textit{permutation tests}.
The conditional expectation $\mu_j \in \R^{p_jq}$ and covariance $\Sigma_j
\in \R^{p_jq \times p_jq}$ of $\T_j(\LS, \w)$ under $H_0$ given
all permutations $\sigma \in S(\LS, \w)$ of the responses are derived by
\cite{Strasser+Weber:1999}:
\begin{eqnarray}
\mu_j & = & \E(\T_j(\LS, \w) | S(\LS, \w)) =
\vec \left( \left( \sum_{i = 1}^n w_i g_j(X_{ji}) \right) \E(h | S(\LS,
\w))^\top
\right), \nonumber \\
\Sigma_j & = & \V(\T_j(\LS, \w) | S(\LS, \w)) \nonumber \\
& = &
\frac{\ws}{\ws - 1} \V(h | S(\LS, \w)) \otimes
\left(\sum_i w_i g_j(X_{ji}) \otimes w_i g_j(X_{ji})^\top \right) \label{expectcovar}
\\
& - & \frac{1}{\ws - 1} \V(h | S(\LS, \w)) \otimes \left(
\sum_i w_i g_j(X_{ji}) \right)
\otimes \left( \sum_i w_i g_j(X_{ji})\right)^\top
\nonumber
\end{eqnarray}
where $\ws = \sum_{i = 1}^n w_i$ denotes the sum of the case weights,
$\otimes$ is the Kronecker product and the conditional expectation of the
influence function is
\begin{eqnarray*}
\E(h | S(\LS, \w)) = \ws^{-1} \sum_i w_i h(\Y_i, (\Y_1, \dots, \Y_n)) \in
\R^q
\end{eqnarray*}
with corresponding $q \times q$ covariance matrix
\begin{eqnarray*}
\V(h | S(\LS, \w)) = \ws^{-1} \sum_i w_i \left(h(\Y_i, (\Y_1, \dots, \Y_n)) - \E(h | S(\LS, \w))
\right) \\
\left(h(\Y_i, (\Y_1, \dots, \Y_n)) - \E(h | S(\LS, \w))\right)^\top.
\end{eqnarray*}
Having the conditional expectation and covariance at hand we are able to
standardize a linear statistic $\T \in \R^{pq}$ of the form
(\ref{linstat}) for some $p \in \{p_1, \dots, p_m\}$.
Univariate test statistics~$c$ mapping an observed multivariate
linear statistic $\bft \in
\R^{pq}$ into the real line can be of arbitrary form. An obvious choice is
the maximum of the absolute values of the standardized linear statistic
\begin{eqnarray*}
c_\text{max}(\bft, \mu, \Sigma) = \max_{k = 1, \dots, pq} \left| \frac{(\bft -
\mu)_k}{\sqrt{(\Sigma)_{kk}}} \right|
\end{eqnarray*}
utilizing the conditional expectation $\mu$ and covariance matrix
$\Sigma$. The application of a quadratic form $c_\text{quad}(\bft, \mu, \Sigma) =
(\bft - \mu) \Sigma^+ (\bft - \mu)^\top$ is one alternative, although
computationally more expensive because the Moore-Penrose
inverse $\Sigma^+$ of $\Sigma$ is involved.
The type of test statistic to be used can be specified by means of the
\code{ctree\_control} function, for example
<>=
ctree_control(teststat = "max")
@
uses $c_\text{max}$ and
<>=
ctree_control(teststat = "quad")
@
takes $c_\text{quad}$ (the default).
It is important to note that the test statistics $c(\bft_j, \mu_j, \Sigma_j),
j = 1, \dots, m$, cannot be directly compared in an unbiased
way unless all of the covariates are measured at the same scale, i.e.,
$p_1 = p_j, j = 2, \dots, m$.
In order to allow for an unbiased variable selection we need to switch to
the $P$-value scale because $P$-values for the conditional distribution of test
statistics $c(\T_j(\LS, \w), \mu_j, \Sigma_j)$
can be directly compared among covariates
measured at different scales. In step 1 of the generic algorithm we
select the covariate with minimum $P$-value, i.e., the covariate $X_{j^*}$
with $j^* = \argmin_{j = 1, \dots, m} P_j$, where
\begin{displaymath}
P_j = \Prob_{H_0^j}(c(\T_j(\LS, \w), \mu_j,
\Sigma_j) \ge c(\bft_j, \mu_j, \Sigma_j) | S(\LS, \w))
\end{displaymath}
denotes the $P$-value of the conditional test for $H_0^j$.
So far, we have only addressed testing each partial hypothesis $H_0^j$, which
is sufficient for an unbiased variable selection. A global test
for $H_0$ required in step 1 can be constructed via an aggregation of the
transformations $g_j, j = 1, \dots, m$,
i.e., using a linear statistic of the form
\begin{eqnarray*}
\T(\LS, \w) = \vec \left( \sum_{i=1}^n w_i
\left(g_1(X_{1i})^\top, \dots,
g_m(X_{mi})^\top\right)^\top h(\Y_i, (\Y_1, \dots, \Y_n))^\top \right).
%%\in \R^{\sum_j p_jq}
\end{eqnarray*}
However, this approach is less attractive for learning samples with
missing values. Universally applicable approaches are multiple test
procedures based on $P_1, \dots, P_m$.
Simple Bonferroni-adjusted $P$-values (the adjustment $1 - (1 - P_j)^m$ is used), available via
<>=
ctree_control(testtype = "Bonferroni")
@
or a min-$P$-value resampling approach (Note: resampling is currently
not implemented in \code{partykit::ctree})
%<>=
%party:::ctree_control(testtype = "MonteCarlo")
%@
are just examples and we refer
to the multiple testing literature \citep[e.g.,][]{Westfall+Young:1993}
for more advanced methods. We reject $H_0$ when the minimum
of the adjusted $P$-values is less than a pre-specified nominal level $\alpha$
and otherwise stop the algorithm. In this sense, $\alpha$
may be seen as a unique parameter determining the size of the resulting trees.
\subsection{Splitting criteria}
Once we have selected a covariate in step 1 of the algorithm, the split itself can be
established by any split criterion, including those established by
\cite{Breiman+Friedman+Olshen:1984} or \cite{Shih:1999}. Instead of simple binary splits,
multiway splits can be implemented as well, for example utilizing
the work of \cite{OBrien:2004}. However, most splitting criteria are not
applicable to response variables measured at arbitrary scales and we
therefore utilize the permutation test framework described above
to find the optimal binary split in one selected
covariate $X_{j^*}$ in step~2 of the generic algorithm. The goodness of a split is
evaluated by two-sample linear statistics which are special cases of the linear
statistic (\ref{linstat}).
For all possible subsets $A$ of the sample space $\sX_{j^*}$ the linear
statistic
\begin{eqnarray*}
\T^A_{j^*}(\LS, \w) = \vec \left( \sum_{i=1}^n w_i I(X_{j^*i} \in A)
h(\Y_i, (\Y_1, \dots, \Y_n))^\top \right) \in \R^{q}
\end{eqnarray*}
induces a two-sample statistic measuring the discrepancy between the samples
$\{ \Y_i | w_i > 0 \text{ and } X_{ji} \in A; i = 1, \dots, n\}$
and $\{ \Y_i | w_i > 0 \text{ and } X_{ji} \not\in A; i = 1, \dots, n\}$.
The conditional expectation $\mu_{j^*}^A$ and covariance
$\Sigma_{j^*}^A$
can be computed by (\ref{expectcovar}).
The split $A^*$ with a test statistic maximized over all possible
subsets $A$ is established:
\begin{eqnarray} \label{split}
A^* = \argmax_A c(\bft_{j^*}^A, \mu_{j^*}^A, \Sigma_{j^*}^A).
\end{eqnarray}
The statistics $c(\bft_{j^*}^A, \mu_{j^*}^A, \Sigma_{j^*}^A)$ are available
for each node with
%<>=
%party:::ctree_control(savesplitstats = TRUE)
%@
and can be used to depict a scatter plot of the covariate
$\sX_{j^*}$ against the statistics (Note: this feature is
currently not implemented in \pkg{partykit}).
Note that we do not need to compute the distribution of
$c(\bft_{j^*}^A, \mu_{j^*}^A, \Sigma_{j^*}^A)$ in step~2.
In order to anticipate pathological splits one can restrict the number of
possible subsets that are evaluated, for example by introducing restrictions
on the sample size or the sum of the case weights in each of the two groups
of observations induced by a possible split. For example,
<>=
ctree_control(minsplit = 20)
@
requires the sum of the weights in both the left and right daughter
node to exceed the value of $20$.
\subsection{Missing values and surrogate splits}
If an observation $X_{ji}$ in covariate $X_j$ is missing, we set the
corresponding case weight $w_i$ to zero for the computation of $\T_j(\LS, \w)$
and, if we would like to split in $X_j$, in $\T_{j}^A(\LS, \w)$ as well.
Once a split $A^*$ in $X_j$ has been implemented, surrogate splits can be
established by
searching for a split leading to roughly the same division of the
observations as the original split. One simply replaces the original
response variable by a binary variable $I(X_{ji} \in A^*)$ coding the
split and proceeds as described in the previous part. The number of
surrogate splits can be controlled using
<>=
ctree_control(maxsurrogate = 3)
@
\subsection{Fitting and inspecting a tree}
For the sake of simplicity, we use a learning sample
<>=
ls <- data.frame(y = gl(3, 50, labels = c("A", "B", "C")),
x1 = rnorm(150) + rep(c(1, 0, 0), c(50, 50, 50)),
x2 = runif(150))
@
in the following illustrations. In \code{partykit::ctree}, the dependency structure and the variables may be
specified in a traditional formula based way
<>=
library("partykit")
ctree(y ~ x1 + x2, data = ls)
@
Case counts $\w$ may be specified using the \code{weights} argument.
Once we have fitted a conditional tree via
<>=
ct <- ctree(y ~ x1 + x2, data = ls)
@
we can inspect the results via a \code{print} method
<>=
ct
@
or by looking at a graphical representation as in Figure~\ref{party-plot1}.
\begin{figure}[t!]
\centering
<>=
plot(ct)
@
\caption{A graphical representation of a classification tree.
\label{party-plot1}}
\end{figure}
Each node can be extracted by its node number, i.e., the root node is
<>=
ct[1]
@
This object is an object of class
<>=
class(ct[1])
@
and we refer to the manual pages for a description of those elements.
The \code{predict} function computes predictions in the space of
the response variable, in our case a factor
<>=
predict(ct, newdata = ls)
@
When we are interested in properties of the conditional distribution of the
response given the covariates, we use
<>=
predict(ct, newdata = ls[c(1, 51, 101),], type = "prob")
@
which, in our case, is a data frame with conditional class probabilities.
We can determine the node numbers of nodes some new observations are falling
into by
<>=
predict(ct, newdata = ls[c(1,51,101),], type = "node")
@
Finally, the \code{sctest} method can be used to extract the test statistics
and $p$-values computed in each node. The function \code{sctest} is used
because for the \code{mob} algorithm such a method (for \underline{s}tructural
\underline{c}hange \underline{test}s) is also provided. To make the generic
available, the \pkg{strucchange} package needs to be loaded (otherwise
\code{sctest.constparty} would have to be called directly).
<>=
library("strucchange")
sctest(ct)
@
Here, we see that \code{x1} leads to a significant test result in the root
node and is hence used for splitting. In the kid nodes, no more significant
results are found and hence splitting stops. For other data sets, other stopping
criteria might also be relevant (e.g., the sample size restrictions \code{minsplit},
\code{minbucket}, etc.). In case, splitting stops due to these, the test results
may also be \code{NULL}.
\section{Examples} \label{examples}
\subsection{Univariate continuous or discrete regression}
For a univariate numeric response $\Y \in \R$, the most natural influence function
is the identity $h(\Y_i, (\Y_1, \dots, \Y_n)) = \Y_i$.
In case some observations with extremely large or small values have been
observed, a
ranking of the observations may be appropriate:
$h(\Y_i, (\Y_1, \dots, \Y_n)) = \sum_{k=1}^n w_k I(\Y_k \le \Y_i)$ for $i = 1, \dots,
n$.
Numeric covariates can be handled by the identity transformation
$g_{ji}(x) = x$ (ranks are possible, too). Nominal covariates at levels $1,
\dots, K$ are
represented by $g_{ji}(k) = e_K(k)$, the unit vector of length $K$ with
$k$th element being equal to one. Due to this flexibility, special test
procedures like the Spearman test, the
Wilcoxon-Mann-Whitney test or the Kruskal-Wallis test and permutation tests
based on ANOVA statistics or correlation coefficients
are covered by this framework. Splits obtained from (\ref{split}) maximize the
absolute value of the standardized difference between two means of the
values of the influence functions.
For prediction, one is usually interested in an estimate of
the expectation of
the response $\E(\Y | \X = \x)$ in each cell, an estimate can be
obtained by
\begin{eqnarray*}
\hat{\E}(\Y | \X = \x) = \left(\sum_{i=1}^n w_i(\x)\right)^{-1} \sum_{i=1}^n
w_i(\x) \Y_i.
\end{eqnarray*}
\subsection{Censored regression}
The influence function $h$ may be chosen as
Logrank or Savage scores taking censoring into
account and one can proceed as for univariate continuous regression. This is
essentially the approach first published by \cite{Segal:1988}.
An alternative is the weighting scheme suggested by
\cite{Molinaro+Dudiot+VanDerLaan:2003}. A weighted Kaplan-Meier curve for the case
weights $\w(\x)$ can serve as prediction.
\subsection{$J$-class classification}
The nominal response variable at levels $1, \dots, J$
is handled by influence
functions\linebreak $h(\Y_i, (\Y_1, \dots, \Y_n)) = e_J(\Y_i)$. Note that for a
nominal covariate $X_j$ at levels $1, \dots, K$ with
$g_{ji}(k) = e_K(k)$ the
corresponding linear statistic $\T_j$ is a vectorized contingency table.
The conditional class probabilities can be estimated via
\begin{eqnarray*}
\hat{\Prob}(\Y = y | \X = \x) = \left(\sum_{i=1}^n w_i(\x)\right)^{-1}
\sum_{i=1}^n
w_i(\x) I(\Y_i = y), \quad y = 1, \dots, J.
\end{eqnarray*}
\subsection{Ordinal regression}
Ordinal response variables measured at $J$ levels, and ordinal covariates
measured at $K$ levels, are associated with score vectors $\xi \in
\R^J$ and $\gamma \in \R^K$, respectively. Those scores reflect the
`distances' between the levels: If the variable is derived from an
underlying continuous variable, the scores can be chosen as the midpoints
of the intervals defining the levels. The linear statistic is now a linear
combination of the linear statistic $\T_j$ of the form
\begin{eqnarray*}
\M \T_j(\LS, \w) & = & \vec \left( \sum_{i=1}^n w_i \gamma^\top g_j(X_{ji})
\left(\xi^\top h(\Y_i, (\Y_1, \dots, \Y_n)\right)^\top \right)
\end{eqnarray*}
with $g_j(x) = e_K(x)$ and $h(\Y_i, (\Y_1, \dots, \Y_n)) = e_J(\Y_i)$.
If both response and covariate are ordinal, the matrix of coefficients
is given by the Kronecker product of both score vectors $\M = \xi \otimes \gamma \in
\R^{1, KJ}$. In case the response is ordinal only, the matrix of
coefficients $\M$ is a block matrix
\begin{eqnarray*}
\M =
\left(
\begin{array}{ccc}
\xi_1 & & 0 \\
& \ddots & \\
0 & & \xi_1
\end{array} \right|
%%\left.
%% \begin{array}{ccc}
%% \xi_2 & & 0 \\
%% & \ddots & \\
%% 0 & & \xi_2
%% \end{array} \right|
%%\left.
\begin{array}{c}
\\
\hdots \\
\\
\end{array} %%\right.
\left|
\begin{array}{ccc}
\xi_q & & 0 \\
& \ddots & \\
0 & & \xi_q
\end{array}
\right)
%%\in \R^{K, KJ}
%%\end{eqnarray*}
\text{ or } %%and if one covariate is ordered
%%\begin{eqnarray*}
%%\M = \left(
%% \begin{array}{cccc}
%% \gamma & 0 & & 0 \\
%% 0 & \gamma & & \vdots \\
%% 0 & 0 & \hdots & 0 \\
%% \vdots & \vdots & & 0 \\
%% 0 & 0 & & \gamma
%% \end{array}
%% \right)
\M = \text{diag}(\gamma)
%%\in \R^{J, KJ}
\end{eqnarray*}
when one covariate is ordered but the response is not. For both $\Y$ and $X_j$
being ordinal, the corresponding test is known as linear-by-linear association
test \citep{Agresti:2002}. Scores can be supplied to \code{ctree} using the
\code{scores} argument, see Section~\ref{illustrations} for an example.
\subsection{Multivariate regression}
For multivariate responses, the influence function is a combination of
influence functions appropriate for any of the univariate response variables
discussed in the previous paragraphs, e.g., indicators for multiple binary
responses \citep{Zhang:1998,Noh+Song+Park:2004}, Logrank or Savage scores
for multiple failure times
%%\citep[for example tooth loss times, ][]{SuFan2004}
and the original observations or a rank transformation for multivariate regression
\citep{Death:2002}.
\section{Illustrations and applications} \label{illustrations}
In this section, we present regression problems which illustrate the
potential fields of application of the methodology.
Conditional inference trees based on $c_\text{quad}$-type test statistics
using the identity influence function for numeric responses
and asymptotic $\chi^2$ distribution are applied.
For the stopping criterion a simple
Bonferroni correction is used and we follow the usual convention by choosing
the nominal level of the conditional independence tests as $\alpha = 0.05$.
\subsection{Tree pipit abundance}
<>=
data("treepipit", package = "coin")
tptree <- ctree(counts ~ ., data = treepipit)
@
\begin{figure}[t!]
\centering
<>=
plot(tptree, terminal_panel = node_barplot)
@
\caption{Conditional regression tree for the tree pipit data.}
\end{figure}
<>=
p <- info_node(node_party(tptree))$p.value
n <- table(predict(tptree, type = "node"))
@
The impact of certain environmental factors on the population density of the
tree pipit \textit{Anthus trivialis}
%%in Frankonian oak forests
is investigated by \cite{Mueller+Hothorn:2004}.
The occurrence of tree pipits was recorded several times at
$n = 86$ stands which were established on a long environmental gradient.
Among nine environmental factors,
the covariate showing the largest association to the number of tree pipits
is the canopy overstorey $(P = \Sexpr{round(p, 3)})$.
Two groups of stands can be distinguished: Sunny stands with less than $40\%$
canopy overstorey $(n = \Sexpr{n[1]})$ show a
significantly higher density of tree pipits compared to darker stands with more than
$40\%$ canopy overstorey $(n = \Sexpr{n[2]})$.
This result is important for management decisions
in forestry enterprises: Cutting the overstorey with release of old
oaks creates a perfect habitat for this indicator species
of near natural forest environments.
\subsection{Glaucoma and laser scanning images}
<>=
data("GlaucomaM", package = "TH.data")
gtree <- ctree(Class ~ ., data = GlaucomaM)
@
<>=
sp <- split_node(node_party(gtree))$varID
@
Laser scanning images taken from the eye background are expected to serve as
the basis of an automated system for glaucoma diagnosis.
Although prediction is more important in this application
\citep{Mardin+Hothorn+Peters:2003}, a simple
visualization of the regression relationship is useful for comparing the
structures inherent in the learning sample with subject matter knowledge.
For $98$ patients and $98$ controls, matched by age
and gender, $62$ covariates describing the eye morphology are available. The
data is part of the
\pkg{TH.data} package, \url{http://CRAN.R-project.org}).
The first split in Figure~\ref{glaucoma} separates eyes with a volume
above reference less than
$\Sexpr{sp} \text{ mm}^3$ in the inferior part of the
optic nerve head (\code{vari}).
Observations with larger volume are mostly controls, a finding which
corresponds to subject matter knowledge: The volume above reference measures the
thickness of the nerve layer, expected to decrease with a glaucomatous
damage of the optic nerve. Further separation is achieved by the volume
above surface global (\code{vasg}) and the volume above reference in the
temporal part of the optic nerve head (\code{vart}).
\setkeys{Gin}{width=.9\textwidth}
\begin{figure}[p!]
\centering
<>=
plot(gtree)
@
\caption{Conditional inference tree for the glaucoma data. For each inner node, the
Bonferroni-adjusted $P$-values are given, the fraction of glaucomatous eyes
is displayed for each terminal node. \label{glaucoma}}
<>=
plot(gtree, inner_panel = node_barplot,
edge_panel = function(...) invisible(), tnex = 1)
@
\caption{Conditional inference tree for the glaucoma data with the
fraction of glaucomatous eyes displayed for both inner and terminal
nodes. \label{glaucoma-inner}}
\end{figure}
The plot in Figure~\ref{glaucoma} is generated by
<>=
plot(gtree)
@
\setkeys{Gin}{width=\textwidth}
and shows the distribution of the classes in
the terminal nodes. This distribution can be shown for the inner nodes as
well, namely by specifying the appropriate panel generating function
(\code{node\_barplot} in our case), see Figure~\ref{glaucoma-inner}.
<>=
plot(gtree, inner_panel = node_barplot,
edge_panel = function(...) invisible(), tnex = 1)
@
%% TH: split statistics are not saved in partykit
%As mentioned in Section~\ref{framework}, it might be interesting to have a
%look at the split statistics the split point estimate was derived from.
%Those statistics can be extracted from the \code{splitstatistic} element
%of a split and one can easily produce scatterplots against the selected
%covariate. For all three inner nodes of \code{gtree}, we produce such a
%plot in Figure~\ref{glaucoma-split}. For the root node, the estimated split point
%seems very natural, since the process of split statistics seems to have a
%clear maximum indicating that the simple split point model is something
%reasonable to assume here. This is less obvious for nodes $2$ and,
%especially, $3$.
%
%\begin{figure}[t!]
%\centering
%<>=
%cex <- 1.6
%inner <- nodes(gtree, c(1, 2, 5))
%layout(matrix(1:length(inner), ncol = length(inner)))
%out <- sapply(inner, function(i) {
% splitstat <- i$psplit$splitstatistic
% x <- GlaucomaM[[i$psplit$variableName]][splitstat > 0]
% plot(x, splitstat[splitstat > 0], main = paste("Node", i$nodeID),
% xlab = i$psplit$variableName, ylab = "Statistic", ylim = c(0, 10),
% cex.axis = cex, cex.lab = cex, cex.main = cex)
% abline(v = i$psplit$splitpoint, lty = 3)
%})
%@
%\caption{Split point estimation in each inner node. The process of
% the standardized two-sample test statistics for each possible
% split point in the selected input variable is show.
% The estimated split point is given as vertical dotted line.
% \label{glaucoma-split}}
%\end{figure}
The class predictions of the tree for the learning sample (and for new
observations as well) can be computed using the \code{predict} function. A
comparison with the true class memberships is done by
<>=
table(predict(gtree), GlaucomaM$Class)
@
When we are interested in conditional class probabilities, the
\code{predict(, type = "prob")} method must be used. A graphical representation is
shown in Figure~\ref{glaucoma-probplot}.
\setkeys{Gin}{width=.5\textwidth}
\begin{figure}[t!]
\centering
<>=
prob <- predict(gtree, type = "prob")[,1] +
runif(nrow(GlaucomaM), min = -0.01, max = 0.01)
splitvar <- character_split(split_node(node_party(gtree)),
data = data_party(gtree))$name
plot(GlaucomaM[[splitvar]], prob,
pch = as.numeric(GlaucomaM$Class), ylab = "Conditional Class Prob.",
xlab = splitvar)
abline(v = split_node(node_party(gtree))$breaks, lty = 2)
legend(0.15, 0.7, pch = 1:2, legend = levels(GlaucomaM$Class), bty = "n")
@
\caption{Estimated conditional class probabilities (slightly jittered)
for the Glaucoma data depending on the first split variable.
The vertical line denotes the first split point. \label{glaucoma-probplot}}
\end{figure}
\subsection{Node positive breast cancer}
Recursive partitioning for censored responses has attracted a lot of interest
\citep[e.g.,][]{Segal:1988,LeBlanc+Crowley:1992}.
Survival trees using $P$-value adjusted Logrank statistics
are used by \cite{Schumacher+Hollaender+Schwarzer:2001a}
for the evaluation of prognostic factors for the German Breast
Cancer Study Group (GBSG2) data, a prospective controlled clinical trial on the
treatment of node positive breast cancer patients. Here, we use
Logrank scores as well. Complete data of seven prognostic factors of $686$
women are used for prognostic modeling, the
dataset is available within the
\pkg{TH.data} package.
The number of positive lymph nodes
(\code{pnodes}) and the progesterone receptor (\code{progrec}) have
been identified as prognostic factors in the survival tree analysis
by \cite{Schumacher+Hollaender+Schwarzer:2001a}. Here, the binary variable coding whether a
hormonal therapy was applied or not (\code{horTh}) additionally is
part of the model depicted in Figure~\ref{gbsg2}, which was
fitted using the following code:
<>=
data("GBSG2", package = "TH.data")
library("survival")
(stree <- ctree(Surv(time, cens) ~ ., data = GBSG2))
@
\setkeys{Gin}{width=\textwidth}
\begin{figure}[t!]
\centering
<>=
plot(stree)
@
\caption{Tree-structured survival model for the GBSG2 data and the distribution of
survival times in the terminal nodes. The median survival time is displayed in
each terminal node of the tree. \label{gbsg2}}
\end{figure}
The estimated median survival time for new patients is less informative
compared to the whole Kaplan-Meier curve estimated from the patients in the
learning sample for each terminal node. We can compute those `predictions'
by means of the \code{treeresponse} method
<>=
pn <- predict(stree, newdata = GBSG2[1:2,], type = "node")
n <- predict(stree, type = "node")
survfit(Surv(time, cens) ~ 1, data = GBSG2, subset = (n == pn[1]))
survfit(Surv(time, cens) ~ 1, data = GBSG2, subset = (n == pn[2]))
@
\subsection{Mammography experience}
<>=
data("mammoexp", package = "TH.data")
mtree <- ctree(ME ~ ., data = mammoexp)
@
\setkeys{Gin}{width=.9\textwidth, keepaspectratio=TRUE}
\begin{figure}[t!]
\centering
<>=
plot(mtree)
@
\caption{Ordinal regression for the mammography experience data with the
fractions of (never, within a year, over one year) given in the nodes.
No admissible split was found for node 5 because only $5$ of $91$ women reported
a family history of breast cancer and the sample size restrictions would
require more than $5$ observations in each daughter node. \label{mammoexp}}
\end{figure}
Ordinal response variables are common in investigations where the response
is a subjective human interpretation.
We use an example given by \cite{Hosmer+Lemeshow:2000}, p.~264,
studying the relationship between the mammography experience (never,
within a year, over one year) and opinions about mammography expressed in
questionnaires answered by $n = 412$ women. The resulting partition based on
scores $\xi = (1,2,3)$ is given in Figure~\ref{mammoexp}.
Women who (strongly) agree with the question `You do not need a mammogram unless
you develop symptoms' seldomly have experienced a mammography. The variable
\code{benefit} is a score with low values indicating a strong agreement with the
benefits of the examination. For those women in (strong) disagreement with the first
question above, low values of \code{benefit} identify persons being more likely to have
experienced such an examination at all.
\subsection{Hunting spiders}
Finally, we take a closer look at a challenging dataset on animal abundance
first reported by \cite{VanDerAart+SmeenkEnserink:1975} and
re-analyzed by \cite{Death:2002} using regression trees dealing with
multivariate responses. The
abundance of $12$ hunting spider species is regressed on six
environmental variables (\code{water}, \code{sand}, \code{moss},
\code{reft}, \code{twigs} and \code{herbs}) for $n = 28$
observations. Because of the small sample size we allow for a split if at
least $5$ observations are element of a node The prognostic factor \code{water}
found by \cite{Death:2002} is confirmed by the model
shown in Figures~\ref{spider1} and~\ref{spider2} which additionally identifies \code{reft}.
The data are available in package
\pkg{mvpart} \citep{mvpart}.
<>=
data("HuntingSpiders", package = "partykit")
sptree <- ctree(arct.lute + pard.lugu + zora.spin + pard.nigr +
pard.pull + aulo.albi + troc.terr + alop.cune + pard.mont + alop.acce +
alop.fabr + arct.peri ~ herbs + reft + moss + sand + twigs + water,
data = HuntingSpiders, teststat = "max", minsplit = 5,
pargs = GenzBretz(abseps = .1, releps = .1))
@
\setkeys{Gin}{width=\textwidth, keepaspectratio=TRUE}
\begin{figure}[t!]
\centering
<>=
plot(sptree, terminal_panel = node_barplot)
@
\caption{Regression tree for hunting spider abundance with bars for the mean of each response. \label{spider1}}
\end{figure}
\setkeys{Gin}{height=.93\textheight, keepaspectratio=TRUE}
\begin{figure}[p!]
\centering
<>=
plot(sptree)
@
\caption{Regression tree for hunting spider abundance with boxplots for each response. \label{spider2}}
\end{figure}
\section{Backward compatibility and novel functionality} \label{sec:novel}
\code{partykit::ctree} is a complete reimplementation of
\code{party::ctree}. The latter reference implementation is based on a
monolithic \proglang{C} core and an \proglang{S4}-based \proglang{R}
interface. The novel implementation of conditional inference trees in
\pkg{partykit} is much more modular and was almost entirely written in
\proglang{R} (package \pkg{partykit} does not contain any foreign language
code as of version 1.2-0). Permutation tests are computed in the dedicated
\proglang{R} add-on package \pkg{libcoin}.
Nevertheless, both implementations will almost every time produce the same
tree. There are, naturally, exceptions where ensuring
backward-compatibility requires specific choices of hyper parameters in
\code{partykit::ctree_control}. We will demonstrate how one can compute
the same trees in \pkg{partykit} and \pkg{party} in this section. In
addition, some novel features introduced in \pkg{partykit} 1.2-0 are
described.
\subsection{Regression}
<>=
library("party")
set.seed(290875)
@
We use the \code{airquality} data from package \pkg{party} and fit a
regression tree after removal of missing response values. There are missing
values in one of the explanatory variables, so we ask for three surrogate
splits to be set-up:
<>=
data("airquality", package = "datasets")
airq <- subset(airquality, !is.na(Ozone))
(airct_party <- party::ctree(Ozone ~ ., data = airq,
controls = party::ctree_control(maxsurrogate = 3)))
mean((airq$Ozone - predict(airct_party))^2)
@
For this specific example, the same call produces the same tree under both
\pkg{party} and \pkg{partykit}. To ensure this also for other patterns of
missingness, the \code{numsurrogate} flag needs to be set in order to
restrict the evaluation of surrogate splits to numeric variables only (this
is a restriction hard-coded in \pkg{party}):
<>=
(airct_partykit <- partykit::ctree(Ozone ~ ., data = airq,
control = partykit::ctree_control(maxsurrogate = 3,
numsurrogate = TRUE)))
mean((airq$Ozone - predict(airct_partykit))^2)
table(predict(airct_party, type = "node"),
predict(airct_partykit, type = "node"))
max(abs(predict(airct_party) - predict(airct_partykit)))
@
The results are identical as are the underlying test statistics:
<>=
airct_party@tree$criterion
info_node(node_party(airct_partykit))
@
\code{partykit} has a nicer way or presenting the variable selection test
statistics on the scale of the statistics and the $p$-values. In addition,
the criterion to be maximised (here: $\log(1 - p-\text{value})$) is given.
\subsection{Classification}
For classification tasks with more than two classes, the default in
\pkg{party} is a maximum-type test statistic on the multidimensional test
statistic when computing splits. \pkg{partykit} employs a quadratic test
statistic by default, because it was found to produce better splits
empirically. One can switch-back to the old behaviour using the
\code{splitstat} argument:
<>=
(irisct_party <- party::ctree(Species ~ .,data = iris))
(irisct_partykit <- partykit::ctree(Species ~ .,data = iris,
control = partykit::ctree_control(splitstat = "maximum")))
table(predict(irisct_party, type = "node"),
predict(irisct_partykit, type = "node"))
@
The interface for computing conditional class probabilities changed from
<>=
tr_party <- treeresponse(irisct_party, newdata = iris)
@
to
<>=
tr_partykit <- predict(irisct_partykit, type = "prob",
newdata = iris)
max(abs(do.call("rbind", tr_party) - tr_partykit))
@
leading to identical results. For ordinal regression, the conditional class
probabilities can be computed in the very same way:
<>=
### ordinal regression
data("mammoexp", package = "TH.data")
(mammoct_party <- party::ctree(ME ~ ., data = mammoexp))
### estimated class probabilities
tr_party <- treeresponse(mammoct_party, newdata = mammoexp)
(mammoct_partykit <- partykit::ctree(ME ~ ., data = mammoexp))
### estimated class probabilities
tr_partykit <- predict(mammoct_partykit, newdata = mammoexp, type = "prob")
max(abs(do.call("rbind", tr_party) - tr_partykit))
@
\subsection{Survival Analysis}
Like in classification analysis, the \code{treeresponse} function from
package \code{party} was replaced by the \code{predict} function with
argument \code{type = "prob"} in \pkg{partykit}. The default survival trees
are identical:
<>=
data("GBSG2", package = "TH.data")
(GBSG2ct_party <- party::ctree(Surv(time, cens) ~ .,data = GBSG2))
(GBSG2ct_partykit <- partykit::ctree(Surv(time, cens) ~ .,data = GBSG2))
@
as are the conditional Kaplan-Meier estimators
<>=
tr_party <- treeresponse(GBSG2ct_party, newdata = GBSG2)
tr_partykit <- predict(GBSG2ct_partykit, newdata = GBSG2, type = "prob")
all.equal(lapply(tr_party, function(x) unclass(x)[!(names(x) %in% "call")]),
lapply(tr_partykit, function(x) unclass(x)[!(names(x) %in% "call")]),
check.names = FALSE)
@
\subsection{New Features}
\pkg{partykit} comes with additional arguments in \code{ctree_control}
allowing a more detailed control over the tree growing.
\begin{description}
\item[\code{alpha}]: The user can optionally change the default nominal
level of $\alpha = 0.05$; \code{mincriterion} is updated to $1 - \alpha$ and
\code{logmincriterion} is then $\log(1 - \alpha)$. The latter allows
variable selection on the scale of $\log(1 - p\text{-value})$:
<>=
(airct_partykit_1 <- partykit::ctree(Ozone ~ ., data = airq,
control = partykit::ctree_control(maxsurrogate = 3, alpha = 0.001,
numsurrogate = FALSE)))
depth(airct_partykit_1)
mean((airq$Ozone - predict(airct_partykit_1))^2)
@
Lower values of $\alpha$ lead to smaller trees.
\item[\code{splittest}]: This enables the computation of $p$-values for
maximally selected statistics for variable selection. The default test
statistic is not particularly powerful against cutpoint-alternatives but much
faster to compute. Currently, $p$-value approximations are not available, so
one has to rely on resampling for $p$-value estimation
<>=
(airct_partykit <- partykit::ctree(Ozone ~ ., data = airq,
control = partykit::ctree_control(maxsurrogate = 3, splittest = TRUE,
testtype = "MonteCarlo")))
@
\item[\code{saveinfo}]: Reduces the memory footprint by not storing test
results as part of the tree. The core information about trees is then
roughly half the size needed by \code{party}.
\item[\code{nmax}]: Restricts the number of possible cutpoints to
\code{nmax}, basically by treating all explanatory variables as ordered
factors defined at quantiles of underlying numeric variables. This is mainly
implemented in package \pkg{libcoin}. For the standard \code{ctree}, it is
only appropriate to use in classification problems, where is can lead to
substantial speed-ups:
<>=
(irisct_partykit_1 <- partykit::ctree(Species ~ .,data = iris,
control = partykit::ctree_control(splitstat = "maximum", nmax = 25)))
table(predict(irisct_partykit), predict(irisct_partykit_1))
@
\item[\code{multiway}]: Implements multiway splits in unordered factors,
each level defines a corresponding daughter node:
<>=
GBSG2$tgrade <- factor(GBSG2$tgrade, ordered = FALSE)
(GBSG2ct_partykit <- partykit::ctree(Surv(time, cens) ~ tgrade,
data = GBSG2, control = partykit::ctree_control(multiway = TRUE,
alpha = .5)))
@
\item[\code{majority = FALSE}]: enables random assignment of non-splitable
observations to daughter nodes preserving the node distribution. With
\code{majority = TRUE}, these observations go with the majority (the only
available behaviour of in \code{party::ctree}).
\end{description}
Two arguments of \code{ctree} are also interesting. The novel \code{cluster}
argument allows conditional inference trees to be fitted to (simple forms
of) correlated observations. For each cluster, the variance of the test
statistics used for variable selection and also splitting is computed
separately, leading to stratified permutation tests (in the sense that only
observations within clusters are permuted). For example, we can cluster the
data in the \code{airquality} dataset by month to be used as cluster
variable:
<>=
airq$month <- factor(airq$Month)
(airct_partykit_3 <- partykit::ctree(Ozone ~ Solar.R + Wind + Temp, data = airq,
cluster = month, control = partykit::ctree_control(maxsurrogate = 3)))
info_node(node_party(airct_partykit_3))
mean((airq$Ozone - predict(airct_partykit_3))^2)
@
This reduces the number of partitioning variables and makes multiplicity
adjustment less costly.
The \code{ytrafo} argument has be made more general. \pkg{party} is not able
to update influence functions $h$ within nodes. With the novel formula-based
interface, users can create influence functions which are newly evaluated
in each node. The following example illustrates how one can compute a
survival tree with updated logrank scores:
<>=
### with weight-dependent log-rank scores
### log-rank trafo for observations in this node only (= weights > 0)
h <- function(y, x, start = NULL, weights, offset, estfun = TRUE, object = FALSE, ...) {
if (is.null(weights)) weights <- rep(1, NROW(y))
s <- logrank_trafo(y[weights > 0,,drop = FALSE])
r <- rep(0, length(weights))
r[weights > 0] <- s
list(estfun = matrix(as.double(r), ncol = 1), converged = TRUE, unweighted = TRUE)
}
partykit::ctree(Surv(time, cens) ~ ., data = GBSG2, ytrafo = h)
@
The results are usually not very sensitive to (simple) updated influence
functions. However, when one uses score functions of more complex models as
influence functions (similar to the \code{mob} family of trees), it is
necessary to refit models in each node. For example, we are interested in a
normal linear model for ozone concentration given temperature; both the
intercept and the regression coefficient for temperature shall vary across
nodes of a tree. Such a ``permutation-based'' MOB, here taking clusters into
account, can be set-up using
<>=
### normal varying intercept / varying coefficient model (aka "mob")
h <- function(y, x, start = NULL, weights = NULL, offset = NULL, cluster = NULL, ...)
glm(y ~ 0 + x, family = gaussian(), start = start, weights = weights, ...)
(airct_partykit_4 <- partykit::ctree(Ozone ~ Temp | Solar.R + Wind,
data = airq, cluster = month, ytrafo = h,
control = partykit::ctree_control(maxsurrogate = 3)))
airq$node <- factor(predict(airct_partykit_4, type = "node"))
summary(m <- glm(Ozone ~ node + node:Temp - 1, data = airq))
mean((predict(m) - airq$Ozone)^2)
@
Both intercept and effect of temperature change considerably between nodes.
The corresponding MOB can be fitted using
<>=
airq_lmtree <- partykit::lmtree(Ozone ~ Temp | Solar.R + Wind,
data = airq, cluster = month)
info_node(node_party(airq_lmtree))
mean((predict(airq_lmtree, newdata = airq) - airq$Ozone)^2)
@
The $p$-values in the root node are similar but the two procedures find
different splits. \code{mob} (and therefore \code{lmtree}) directly search
for splits by optimising the objective function for all possible splits
whereas \code{ctree} only works with the score functions.
Argument \code{xtrafo} allowing the user to change the transformations $g_j$ of the
covariates was removed from the user interface.
<>=
detach(package:party)
@
\bibliography{party}
\end{document}