feat(ets): add more params

R/LearnerRegrEts.R

@@ -20,7 +20,26 @@ LearnerFcstEts = R6Class("LearnerFcstEts",
     #' @description
     #' Creates a new instance of this [R6][R6::R6Class] class.
     initialize = function() {
-      param_set = ps()
+      param_set = ps(
+        model = p_uty(default = "ZZZ", tags = "train", custom_check = crate(function(x) check_string(x, n.chars = 3L))),
+        damped = p_lgl(default = NULL, special_vals = list(NULL), tags = "train"),
+        alpha = p_dbl(default = NULL, special_vals = list(NULL), tags = "train"),
+        beta = p_dbl(default = NULL, special_vals = list(NULL), tags = "train"),
+        gamma = p_dbl(default = NULL, special_vals = list(NULL), tags = "train"),
+        phi = p_dbl(default = NULL, special_vals = list(NULL), tags = "train"),
+        additive.only = p_lgl(default = FALSE, tags = "train"),
+        lambda = p_uty(tags = "train"),
+        biasadj = p_lgl(default = FALSE, tags = "train"),
+        lower = p_uty(default = c(rep(1e-04, 3), 0.8), tags = "train"),
+        upper = p_uty(default = c(rep(0.9999, 3), 0.98), tags = "train"),
+        opt.crit = p_fct(default = "lik", levels = c("lik", "amse", "mse", "sigma", "mae"), tags = "train"),
+        nmse = p_int(0L, 30L, default = 3, tags = "train"),
+        bounds = p_fct(default = "both", levels = c("both", "usual", "admissible"), tags = "train"),
+        ic = p_fct(default = "aicc", levels = c("aicc", "aic", "bic"), tags = "train"),
+        restrict = p_lgl(default = TRUE, tags = "train"),
+        allow.multiplicative.trend = p_lgl(default = FALSE, tags = "train"),
+        na.action = p_fct(default = "na.contiguous", levels = c("na.contiguous", "na.interp", "na.fail"))
+      )
 
       super$initialize(
         id = "fcst.ets",
@@ -43,6 +62,7 @@ LearnerFcstEts = R6Class("LearnerFcstEts",
       }
       private$.max_index = max(task$data(cols = task$col_roles$order)[[1L]])
       pv = self$param_set$get_values(tags = "train")
+      # TODO: is this relefant for ETS?
       if ("weights" %in% task$properties) {
         pv = insert_named(pv, list(weights = task$weights$weight))
       }