【发布时间】:2018-10-08 08:15:29
【问题描述】:
也许我搜索了错误的关键字以找到解决此问题的方法,但到目前为止我还没有在网上找到答案。
情况:我已经使用 CARET 包在 R 中训练了一个逻辑回归模型,该模型具有正则化,用于二类分类。然后,我使用该模型预测了我的训练和验证集的类别,以接收混淆矩阵并衡量模型的性能。
问题:现在我想根据我的模型识别具有错误类的对象。因此,那些在实际上属于“否”类时被预测为“是”的那些,在混淆矩阵中反之亦然。我想这样做来检查对象,以改进或更改功能以获得更好的模型。我想知道的是它们在数据框中的位置。
我的相关代码:
set.seed(1)
datapart <- createDataPartition(m$class, p = .85, list = FALSE)
train <- m[datapart ,]
validation <- m[-datapart ,]
library(glmnet)
set.seed(1)
glmnetcontrol <- trainControl(method="repeatedcv", number=10, repeats=3, classProbs = TRUE, summaryFunction = twoClassSummary, search = "random")
glmnet <- train(class~., data=train, method="glmnet", trControl = glmnetcontrol, metric="ROC")
predict_validation <- predict(glmnet, validation, type="raw")
predict_train <- predict(glmnet, train, type="raw")
quality_glmnet_valid <- confusionMatrix(predict_validation , validation$class)
quality_glmnet_train <- confusionMatrix(predict_train , train$class)
【问题讨论】:
-
您可以尝试
str(quality_glmnet_valid)检查返回的内容并使用$运算符访问您想要的元素
标签: r classification r-caret