set.seed(1)

low_birth_knn = train(Low ~ .-Last_Menstruation_Weight_bins, method = "knn", data = train_low_birth,
                   trControl = trainControl(method = 'cv', number = 5))

set.seed(1)

predicted_outcomes <- predict(low_birth_knn, test_low_birth)

knn_confM <- confusionMatrix(predicted_outcomes, test_low_birth$Low)

set.seed(1)

low_birth_knn_race = train(Low ~ (.-Last_Menstruation_Weight_bins -Race), method = "knn", data = train_low_birth,
                      trControl = trainControl(method = 'cv', number = 5))

set.seed(1)

predicted_outcomes_race <- predict(low_birth_knn_race, test_low_birth)

knn_confM_race <- confusionMatrix(predicted_outcomes_race, test_low_birth$Low)