mirror of
https://github.com/saymrwulf/prophet.git
synced 2026-07-27 20:02:18 +00:00
This commit is contained in:
parent
16b0166518
commit
e9c017ba01
2 changed files with 2 additions and 0 deletions
|
|
@ -1257,6 +1257,7 @@ fit.prophet <- function(m, df, ...) {
|
|||
object = model,
|
||||
data = dat,
|
||||
init = stan_init,
|
||||
algorithm = if(dat$T < 100) {'Newton'} else {'LBFGS'},
|
||||
iter = 1e4,
|
||||
as_vector = FALSE
|
||||
)
|
||||
|
|
|
|||
|
|
@ -1114,6 +1114,7 @@ class Prophet(object):
|
|||
args = dict(
|
||||
data=dat,
|
||||
init=stan_init,
|
||||
algorithm='Newton' if dat['T'] < 100 else 'LBFGS',
|
||||
iter=1e4,
|
||||
)
|
||||
args.update(kwargs)
|
||||
|
|
|
|||
Loading…
Reference in a new issue