Skip to content

Commit 031bd0a

Browse files
m-mueckem-muecke
committed
chore: more arima params
1 parent 95a3df7 commit 031bd0a

File tree

2 files changed

+26
-11
lines changed

2 files changed

+26
-11
lines changed

‎R/LearnerFcstArima.R‎

Lines changed: 11 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,17 @@ LearnerFcstArima = R6Class(
3838
include.constant = p_lgl(default = FALSE, tags = "train"),
3939
lambda = p_uty(default = NULL, tags = "train"),
4040
biasadj = p_lgl(default = FALSE, tags = "train"),
41-
method = p_fct(c("CSS-ML", "ML", "CSS"), default = "CSS-ML", tags = "train")
41+
method = p_fct(c("CSS-ML", "ML", "CSS"), default = "CSS-ML", tags = "train"),
42+
# additional arguments to stats::arima
43+
transform.pars = p_lgl(default = TRUE, tags = "train"),
44+
SSinit = p_fct(c("Gardner1980", "Rossignol2011"), default = "Gardner1980", tags = "train"),
45+
optim.method = p_fct(
46+
c("Nelder-Mead", "BFGS", "CG", "L-BFGS-B", "SANN", "Brent"),
47+
default = "BFGS",
48+
tags = "train"
49+
),
50+
optim.control = p_uty(default = list(), tags = "train", custom_check = check_list),
51+
kappa = p_dbl(default = 1e6, tags = "train")
4252
)
4353

4454
super$initialize(

‎man/mlr_learners_fcst.arima.Rd‎

Lines changed: 15 additions & 10 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

0 commit comments

Comments
 (0)