Já používám stříška balíčky porovnat různé modely.
Po tréninku model, jak najít ROC prostoru.
# Split data
a<- createDataPartition(data$target, p = .8, list = FALSE)
train <- data[ a,]
test <- data[-a,]
myControl = trainControl(
method = "cv",
summaryFunction = twoClassSummary,
classProbs = TRUE,
verboseIter = FALSE,
)
model_knn = train(
target ~ .,
train,
method = "knn",
metric = "ROC",
tuneLength = 10,
trControl = myControl)
Například, to je jeden z modelů postaven. Pokud jsem se udělat následující, můžu ROC křivka můj trénink nastavit. Ale dostat ROC mého testování sada dat?
model_knn
plot(model)