Survival models make indicators

Author: juliasilge

R/surv_reg_data.R

@@ -29,6 +29,13 @@ set_fit(
   )
 )
 
+set_encoding(
+  model = "surv_reg",
+  eng = "flexsurv",
+  mode = "regression",
+  options = list(predictor_indicators = TRUE)
+)
+
 set_pred(
   model = "surv_reg",
   eng = "flexsurv",
@@ -92,6 +99,13 @@ set_fit(
   )
 )
 
+set_encoding(
+  model = "surv_reg",
+  eng = "survival",
+  mode = "regression",
+  options = list(predictor_indicators = TRUE)
+)
+
 set_pred(
   model = "surv_reg",
   eng = "survival",