r中Logistic回归模型的ConfusionMatrix



我试图在我的逻辑回归模型上计算混淆矩阵,但得到一个错误

错误:数据和参考应为相同级别的因子。另外:警告信息:In is.factor(reference): NAs imported通过强迫

代码:

usage_model_rl <- glm(data = train_usage_rl, formula = Subscription ~., family = binomial(link = "logit"))
usage_predictions <- predict(usage_model_rl, test_usage_rl, type = "response") %>%
bind_cols(test_usage_rl %>% dplyr::select(Subscription), preds = .)

cut04 <- usage_predictions %>%mutate(predicted = ifelse(preds >= 0.4, 1, 0)) %>% dplyr::select(-preds) %>% dplyr::select(predicted, Subscription) %>%mutate_all(list(~ factor(., levels = c(1, 0)))) %>%table()
confusionMatrix(usage_predictions,as.numeric(test_usage$Subscription),positive==1)

试试这个:

confusionMatrix(usage_predictions,test_usage$Subscription,positive="1", dnn=c("Prediction", "Reference"))

确保它们都是因子。

最新更新