removes trend from logistic regression model

nhs-bnssg-analytics · Jan 18, 2024 · 77096d8 · 77096d8
1 parent ecb6620
commit 77096d8
Showing 1 changed file with 10 additions and 10 deletions.
diff --git a/R/04_modelling_utils.R b/R/04_modelling_utils.R
@@ -180,9 +180,9 @@ identify_missing_vars <- function(data_train, data_validation, data_test, model_
     c(mostly_missing_vars, mostly_zero_vars, entirely_missing_vars, "org")
   )
 
-  if (model_type == "random_forest") {
+  # if (model_type == "random_forest") {
     vars_to_remove <- c(vars_to_remove, "year")
-  }
+  # }
 
   # identify fields to impute
   missing_data <- names(data)[colSums(is.na(data)) > 0] |> 
@@ -649,14 +649,14 @@ modelling_performance <- function(data, target_variable, lagged_years = 0,
     )
 
   if (model_type %in% "logistic_regression") {
-    model_recipe <- model_recipe |> 
-      step_rename(
-        years_from_2020 = year
-      ) |> 
-      step_mutate(
-        years_from_2020 = years_from_2020 - 2020,
-        role = "predictor"
-      )
+    # model_recipe <- model_recipe |> 
+    #   step_rename(
+    #     years_from_2020 = year
+    #   ) |> 
+    #   step_mutate(
+    #     years_from_2020 = years_from_2020 - 2020,
+    #     role = "predictor"
+    #   )
   } else if (model_type %in% c("random_forest")) {
     model_recipe <- model_recipe |> 
       step_rm(