if(check_pkg_installed("caret", raise_error = FALSE)) {
df = mtcars2[, ! names(mtcars2) %in% 'ids' ]
train = caret::train( disp ~ .,
df,
method = 'rf',
trControl = caret::trainControl( method = 'none' ),
importance = TRUE )
alluvial_model_response_caret(train, df, degree = 3)
}
# partial dependency plotting method
if (FALSE) {
future::plan("multisession")
alluvial_model_response_caret(train, df, degree = 3, method = 'pdp', parallel = TRUE)
}
Run the code above in your browser using DataLab