prepare_and_fit(): set cohort in DataSpec to train if not specified

lgessl · Jan 12, 2024 · 5465db5 · 5465db5
1 parent ba9356c
commit 5465db5
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 16 deletions.
diff --git a/R/prepare_and_fit.R b/R/prepare_and_fit.R
@@ -31,6 +31,8 @@ prepare_and_fit <- function(
         message("\tCreating ", directory)
         dir.create(directory, recursive = TRUE)
     }
+    if(is.null(data_spec$cohort))
+        data_spec$cohort <- "train"
     # Set up list holding fits
     fits <- list()
     stored_fits_fname <- file.path(model_spec$directory, model_spec$fit_fname)

diff --git a/man/ModelSpec.Rd b/man/ModelSpec.Rd
diff --git a/man/generate_response.Rd b/man/generate_response.Rd