diff --git a/R/LearnerRegrEts.R b/R/LearnerRegrEts.R index 2ec9431..1245ae6 100644 --- a/R/LearnerRegrEts.R +++ b/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)) } diff --git a/man/mlr_learners_fcst.ets.Rd b/man/mlr_learners_fcst.ets.Rd index fc0f3da..60d72fc 100644 --- a/man/mlr_learners_fcst.ets.Rd +++ b/man/mlr_learners_fcst.ets.Rd @@ -28,8 +28,27 @@ lrn("fcst.ets") } \section{Parameters}{ - -Empty ParamSet +\tabular{lllll}{ + Id \tab Type \tab Default \tab Levels \tab Range \cr + model \tab untyped \tab "ZZZ" \tab \tab - \cr + damped \tab logical \tab NULL \tab TRUE, FALSE \tab - \cr + alpha \tab numeric \tab NULL \tab \tab \eqn{(-\infty, \infty)}{(-Inf, Inf)} \cr + beta \tab numeric \tab NULL \tab \tab \eqn{(-\infty, \infty)}{(-Inf, Inf)} \cr + gamma \tab numeric \tab NULL \tab \tab \eqn{(-\infty, \infty)}{(-Inf, Inf)} \cr + phi \tab numeric \tab NULL \tab \tab \eqn{(-\infty, \infty)}{(-Inf, Inf)} \cr + additive.only \tab logical \tab FALSE \tab TRUE, FALSE \tab - \cr + lambda \tab untyped \tab - \tab \tab - \cr + biasadj \tab logical \tab FALSE \tab TRUE, FALSE \tab - \cr + lower \tab untyped \tab c(rep(1e-04, 3), 0.8) \tab \tab - \cr + upper \tab untyped \tab c(rep(0.9999, 3), 0.98) \tab \tab - \cr + opt.crit \tab character \tab lik \tab lik, amse, mse, sigma, mae \tab - \cr + nmse \tab integer \tab 3 \tab \tab \eqn{[0, 30]}{[0, 30]} \cr + bounds \tab character \tab both \tab both, usual, admissible \tab - \cr + ic \tab character \tab aicc \tab aicc, aic, bic \tab - \cr + restrict \tab logical \tab TRUE \tab TRUE, FALSE \tab - \cr + allow.multiplicative.trend \tab logical \tab FALSE \tab TRUE, FALSE \tab - \cr + na.action \tab character \tab na.contiguous \tab na.contiguous, na.interp, na.fail \tab - \cr +} } \references{