fix confusion matrix

This commit is contained in:
2024-12-10 10:01:30 +01:00
parent 332ac9e1e7
commit a1e5fe2d9c

View File

@@ -84,12 +84,11 @@ model_train_pima <- glm(type ~ npreg + glu + bp + skin + bmi + ped + age, data =
summary(model_train_pima)
```
```{r}
pima.te$pred <- predict(model_train_pima, newdata = pima.te, type = "response")
pima.te$pred <- ifelse(pima.te$pred > 0.5, "Yes", "No")
pima.te$pred <- ifelse(pima.te$pred > 0.55, "Yes", "No")
pima.te$pred <- as.factor(pima.te$pred)
pima.te$type <- as.factor(pima.te$type)
# Confusion matrix
confusionMatrix(data = pima.te$type, reference = pima.te$pred)
confusionMatrix(data = pima.te$type, reference = pima.te$pred, positive = 'Yes')
```