From 2922a4d82de9842a293a9997a859c3bde084ad78 Mon Sep 17 00:00:00 2001
From: Marder <fm58hufi@usr.idiv.de>
Date: Tue, 8 Oct 2024 09:37:33 +0200
Subject: [PATCH] matching
---
Scripts/logit/chr_vol_treat.R | 5 +++++
1 file changed, 5 insertions(+)
diff --git a/Scripts/logit/chr_vol_treat.R b/Scripts/logit/chr_vol_treat.R
index 67ddbf8..399ec54 100644
--- a/Scripts/logit/chr_vol_treat.R
+++ b/Scripts/logit/chr_vol_treat.R
@@ -94,6 +94,7 @@ data <- data %>%
# Split the data into labeled and unlabeled sets
labeled_data <- filter(data, Choice_Treat==1| Choice_Treat==0)
unlabeled_data <- filter(data, is.na(Choice_Treat))
+labeled_data_id<-labeled_data
labeled_data<-select(labeled_data,-id)
# Assuming the group information is in the column called 'Group'
labeled_data$Choice_Treat<- as.factor(labeled_data$Choice_Treat)
@@ -140,6 +141,10 @@ labeled_data$PredictedGroup <- labeled_predictions
table(labeled_data$Choice_Treat, labeled_data$PredictedGroup)
unlabeled_predictions <- predict(model3, newdata = unlabeled_data)
+labeled_data_id$PredictedGroup <- labeled_predictions
+data_prediction_labeled<-select(labeled_data_id, c("id", "PredictedGroup"))
+saveRDS(data_prediction_labeled, "Data/predictions_labeled.RDS")
+
unlabeled_data$PredictedGroup <- unlabeled_predictions
data_prediction<-select(unlabeled_data, c("id", "PredictedGroup"))
saveRDS(data_prediction, "Data/predictions.RDS")
--
GitLab