Zug auf Spezifität

Question

Ich benutze Caret, um mein Modell (binäre Klassifizierungsaufgabe) zu trainieren. Wie kann ich sicherstellen, dass train() nicht auf der Genauigkeitsmetrik, sondern auf der Spezifitätsmetrik (TN/(TN + FP)) trainiert?

, was funktioniert auf Genauigkeit:

metric = "Specificity" 

Wer weiß, wie das Modell zu trainieren, um die Spezifität zu optimieren:

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

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

Es ist nicht zu ändern funktioniert?

KR, Arnand

Answer 1

Versuchen Sie das summaryFunction Argument twoClassSummary innen trainControl zusammen mit classProbs = TRUE specifiyng und metric = "Spec" innen train():

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

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