R/analysis_tools.R
split_frequencies.Rd
Calculate which features the forest split on at each depth.
split_frequencies(forest, max.depth = 4)
forest | The trained forest. |
---|---|
max.depth | Maximum depth of splits to consider. |
A matrix of split depth by feature index, where each value is the number of times the feature was split on at that depth.
# \donttest{ # Train a quantile forest. n <- 250 p <- 10 X <- matrix(rnorm(n * p), n, p) Y <- X[, 1] * rnorm(n) q.forest <- quantile_forest(X, Y, quantiles = c(0.1, 0.5, 0.9)) # Calculate the split frequencies for this forest. split_frequencies(q.forest)#> [,1] [,2] [,3] [,4] [,5] [,6] [,7] [,8] [,9] [,10] #> [1,] 793 138 115 123 147 119 242 104 97 122 #> [2,] 856 334 342 316 280 339 401 276 293 319 #> [3,] 960 500 517 478 553 506 478 448 502 498 #> [4,] 932 624 620 576 563 607 639 553 565 578# }