library("DALEX2")
Y_train <- HR$status
x_train <- HR[ , -6]
head(HR)

library("randomForest")
set.seed(123)
model_rf <- randomForest(x = x_train, y = Y_train)

library(shapper)

p_function <- function(model, data) predict(model, newdata = data, type = "prob")

ive_rf <- individual_variable_effect(model_rf, data = x_train, predict_function = p_function,
            new_observation = x_train[1:2,], nsamples = 50)

print(ive_rf)

plot(ive_rf)

Y_train <- dragons$life_length
x_train <- dragons[ , -8]

head(dragons)

set.seed(123)
model_rf <- randomForest(x = x_train, y = Y_train)

ive_rf <- individual_variable_effect(model_rf, data = x_train,
                                     new_observation = x_train[1,])

plot(ive_rf)