Train on Specific

Question

Я использую Caret для обучения моей модели (задача двоичной классификации). Как я могу убедиться, что train() не тренируется по метрике точности, а по метрике Specificity (TN/(TN + FP))?

что работает на точность:

control <- trainControl(method="cv", number=10) 
metric <- "Accuracy" 

set.seed(7) 
fit.svm <- train(target_var ~., data=dataset, method="svmRadial", metric=metric, trControl=control) 

Это не работает для изменения:

metric = "Specificity" 

Кто-нибудь знает, как тренировать модель для оптимизации Специфичность?

KR, Arnand

Answer 1

Попробуйте specifiyng в summaryFunction аргумент twoClassSummary внутри trainControl вместе с classProbs = TRUE и metric = "Spec" внутри train():

control <- trainControl(method="cv", 
         number=10, 
         summaryFunction = twoClassSummary, 
         classProbs = TRUE) 

fit.svm <- train(target_var ~., 
       data=dataset, 
       method="svmRadial", 
       metric="Spec", 
       trControl=control)