leaf_membership {aggTrees} | R Documentation |

## Leaf Membership

### Description

Constructs a variable that encodes in which leaf of an `rpart`

object the units in a given data frame fall.

### Usage

```
leaf_membership(tree, X)
```

### Arguments

`tree` |
An |

`X` |
Covariate matrix (no intercept). |

### Value

A factor whose levels denote in which leaf each unit falls. Leaves are ordered in increasing order of their predictions (from most negative to most positive).

### Author(s)

Riccardo Di Francesco

### See Also

`subtree`

`node_membership`

`get_leaves`

### Examples

```
## Generate data.
set.seed(1986)
n <- 3000
k <- 3
X <- matrix(rnorm(n * k), ncol = k)
colnames(X) <- paste0("x", seq_len(k))
y <- exp(X[, 1]) + 2 * X[, 2] * X[, 2] > 0 + rnorm(n)
## Construct tree.
library(rpart)
tree <- rpart(y ~ ., data = data.frame(y, X))
## Extract number of leaves.
leaves_factor <- leaf_membership(tree, X)
head(leaves_factor)
```

[Package

*aggTrees*version 2.0.2 Index]