inst/tinytest/test_treemisc.R

In bgreenwell/treemisc: Data Sets and Functions to Accompany "Tree-Based Methods for Statistical Learning in R"

# Exits
if (!requireNamespace("AmesHousing", quietly = TRUE)) {
  exit_file("Package 'AmesHousing' missing.")
}
if (!requireNamespace("Matrix", quietly = TRUE)) {
  exit_file("Package 'Matrix' missing.")
}
if (!requireNamespace("ranger", quietly = TRUE)) {
  exit_file("Package 'ranger' missing.")
}

# Load required packages
library(Matrix)
library(ranger)

# Load Ames housing data
ames <- as.data.frame(AmesHousing::make_ames())

# Fit a ranfom forest to the Ames housing data
set.seed(822)  # for reproducibility
rfo <- ranger(Sale_Price ~ ., data = ames)

# Compute proximity matrix
system.time(
  ames_prox <- proximity(rfo, data = ames) 
)

# Check size
format(object.size(ames_prox), units = "MB")

# Visualize proximities
# image(ames_prox)

# mds <- stats::cmdscale(1 - ames_prox, eig = TRUE, k = 2)
# plot(mds$points)