# Draw some data
X1 <- matrix(rnorm(1000), ncol = 10)
X2 <- matrix(rnorm(1000, mean = 0.5), ncol = 10)
y1 <- rbinom(100, 1, 1 / (1 + exp(1 - X1 %*% rep(0.5, 10))))
y2 <- rbinom(100, 1, 1 / (1 + exp(1 - X2 %*% rep(0.7, 10))))
X1 <- data.frame(X = X1, y = y1)
X2 <- data.frame(X = X2, y = y2)
# Calculate Ganti et al. statistic (without tuning and testing due to runtime)
if(requireNamespace("rpart", quietly = TRUE)) {
GGRL(X1, X2, "y", "y", tune = FALSE)
}
# Categorical case
set.seed(1234)
X1 <- data.frame(X1 = factor(sample(letters[1:5], 1000, TRUE)),
X2 = factor(sample(letters[1:4], 1000, TRUE)),
X3 = factor(sample(letters[1:3], 1000, TRUE)),
y = sample(0:1, 100, TRUE))
X2 <- data.frame(X1 = factor(sample(letters[1:5], 1000, TRUE, 1:5)),
X2 = factor(sample(letters[1:4], 1000, TRUE, 1:4)),
X3 = factor(sample(letters[1:3], 1000, TRUE, 1:3)),
y = sample(0:1, 100, TRUE))
# Calculate Ganti et al. statistic (without tuning and testing due to runtime)
if(requireNamespace("rpart", quietly = TRUE)) {
GGRLCat(X1, X2, "y", "y", tune = FALSE)
}
Run the code above in your browser using DataLab