# Load libraries
library(caret)
data(iris)
train_data <- iris[, 1:4]
train_classes <- iris[, 5]
train_fit <- train(train_data, train_classes,
method = "knn",
preProcess = c("center", "scale"),
tuneLength = 10,
trControl = trainControl(method = "cv"))
out <- axe_env(train_fit$modelInfo$prob, verbose = TRUE)
out <- axe_env(train_fit$modelInfo$levels, verbose = TRUE)
out <- axe_env(train_fit$modelInfo$predict, verbose = TRUE)
Run the code above in your browser using DataLab