diff --git a/Scripts/logit/chr_vol_treat.R b/Scripts/logit/chr_vol_treat.R
index 782020e7dd03842f3336738da409696b9634e4f5..6bd1e6bd7199aaa9ab8cfee687e5a14fcef7fc64 100644
--- a/Scripts/logit/chr_vol_treat.R
+++ b/Scripts/logit/chr_vol_treat.R
@@ -141,6 +141,9 @@ table(labeled_data$Choice_Treat, labeled_data$PredictedGroup)
 
 unlabeled_predictions <- predict(model3, newdata = unlabeled_data)
 unlabeled_data$PredictedGroup <- unlabeled_predictions
+data_prediction<-select(unlabeled_data, c("id", "PredictedGroup"))
+saveRDS(data_prediction, "Data/predictions.RDS")
+
 print(model3$bestTune)
 
 
@@ -177,4 +180,7 @@ best_coords <- coords(roc_obj, "best", best.method="youden")
 
 cut_off <- best_coords$threshold
 
+<<<<<<< HEAD
 
+=======
+>>>>>>> refs/remotes/origin/main