diff --git a/Scripts/logit/chr_vol_treat.R b/Scripts/logit/chr_vol_treat.R
index 6b67cc141870388df191f02e7428f3026cd0de91..67ddbf8e9ad8ff22c5b59db968ddd700ec0fc74a 100644
--- a/Scripts/logit/chr_vol_treat.R
+++ b/Scripts/logit/chr_vol_treat.R
@@ -76,7 +76,7 @@ data<-select(data, Choice_Treat, id, Age, Q02W123, Uni_degree, QFIncome,Z_Mean_N
              Uni_degree, Q02W123,Q04W123,Q05W123, Q08W123,Q09W123,Q10W123,Q11W3,Q12W123,Q13W23,Q14S01W123,Q14S02W23,
              Q14S03W123,Q14S04W123,Q14S05W123,Q14S06W23,Q14S07W3,Q14S08W2,Q14W23,
              Q15S01W3,Q15S02W3,Q16W3,Q17W13,Q18W123,Q19W3,C02W23,Q20W23,Q21W23,
-             Q22S01W123,Q22S02W23,Q23W123,Q24S01W123,
+             Q22S01W123,Q22S02W23,UGS_visits,Q24S01W123,
              Q24S02W123,Q24S03W123,Q24S04W23,Q24S05W123,Q25W23,Q26S01W123,Q26S02W123,
              Q26S03W23,Q26S04W123,Q26S05W123,Q26S06W123,Q26S07W23,Q26S08W23,Q26S99W23,
              Q27W123,Q30W23,Q31S01W23,Q31S02W23,Q31S03W23,Q31S04W23,
@@ -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,3 @@ best_coords <- coords(roc_obj, "best", best.method="youden")
 
 cut_off <- best_coords$threshold
 
->>>>>>> e410a7a5c52ecf34b454cff74f0b641cd5615679