U Kv f–'ã@sÖddlZe e e¡j¡ZGdd„dƒZdd„Zdd„Z edœd d „Z edœd d „Z edœd d„Z edœdd„Z edœdd„Zedœdd„Zedœdd„Zedœdd„Zedœdd„Zedœdd„Zedœdd„ZdS)éNc@sªeZdZddd„Zedd„ƒZejdd„ƒZedd„ƒZed d „ƒZed d „ƒZ ed d„ƒZ edd„ƒZ edd„ƒZ edd„ƒZ edd„ƒZedd„ƒZejdd„ƒZdS)ÚHoltWintersArgsFcCsZ||_||_||_||_t |¡|_t |¡|_t ||d¡|_||_ ||_ ||_ dS)Né) Ú_xiÚ_pÚ_boundsÚ_yÚnpÚemptyÚ_lvlÚ_bÚ_sÚ_mÚ_nÚ _transform)ÚselfÚxiÚpÚboundsÚyÚmÚnÚ transform©rúJ/tmp/pip-unpacked-wheel-2v6byqio/statsmodels/tsa/holtwinters/_smoothers.pyÚ__init__s  zHoltWintersArgs.__init__cCs|jS©N©r©rrrrrszHoltWintersArgs.xicCs ||_dSrr©rÚvaluerrrrscCs|jSr)rrrrrrszHoltWintersArgs.pcCs|jSr)rrrrrrszHoltWintersArgs.boundscCs|jSr)rrrrrr#szHoltWintersArgs.ycCs|jSr)r rrrrÚlvl'szHoltWintersArgs.lvlcCs|jSr)r rrrrÚb+szHoltWintersArgs.bcCs|jSr)r rrrrÚs/szHoltWintersArgs.scCs|jSr)r rrrrr3szHoltWintersArgs.mcCs|jSr)rrrrrr7szHoltWintersArgs.ncCs|jSr©rrrrrr;szHoltWintersArgs.transformcCs ||_dSrr#rrrrr?sN)F)Ú__name__Ú __module__Ú __qualname__rÚpropertyrÚsetterrrrr r!r"rrrrrrrrs2            rcCs´|dd…\}}}|drJtt|dƒ}tdt|dƒ}||||}|drx|d}t||dƒ}||||}|d rª|d }td ||d ƒ}||||}|||fS) aÿ Transform parameters from the unrestricted [0,1] space to satisfy both the bounds and the 2 constraints beta <= alpha and gamma <= (1-alpha) Parameters ---------- p : ndarray The parameters to transform sel : ndarray Array indicating whether a parameter is being estimated. If not estimated, not transformed. bounds : ndarray 2-d array of bounds where bound for element i is in row i and stored as [lb, ub] Returns ------- Nér©rrr©rr©rr©rré©r.rçð?©r.r©ÚmaxÚ LOWER_BOUNDÚmin©rÚselrÚar!ÚgZlbZubrrrÚ to_restrictedDsr:cCs¼|dd…\}}}|drJtt|dƒ}tdt|dƒ}||||}|dr||d}t|d|dƒ}||||}|d r²|d }td |d|d ƒ}||||}|||fS) zë Transform parameters to the unrestricted [0,1] space Parameters ---------- p : ndarray Parameters that strictly satisfy the constraints Returns ------- ndarray Parameters all in (0,1) Nr)rr*rr+r,r-r.r/r0r1r2r6rrrÚto_unrestrictedksr;)Úhw_argsc Cs–||j|j t¡<|jr2t|j|j|jƒ\}}}n|jdd…\}}|jdd…\}}}d|}d|} ||j} ||jd<||j d<||||| | fS)z, Initialization for the Holt Models Nr.r)érr) rrÚastypeÚboolrr:rrr r!) Úxr<ÚalphaÚbetaÚ_Úl0Úb0ÚphiÚalphacÚbetacÚy_alpharrrÚ holt_inits   rJcCs\t||ƒ\}}}}}}|j}|j}td|ƒD]$}||d|||d||<q,|j|S)zH Simple Exponential Smoothing Minimization Function (,) r)rJrr Úranger)r@r<rCrGrIrr ÚirrrÚholt__©s "rMc Cs¦t||ƒ\}}}}}}|j}|j} td|jƒD]d} || d||| d| | d||| <||| || d|| | d|| | <q.|j|| |S)z] Multiplicative and Multiplicative Damped Minimization Function (M,) & (Md,) r©rJr r!rKrr© r@r<rCrBrFrGrHrIr r!rLrrrÚ holt_mul_dam·s02rPc Cs¦t||ƒ\}}}}}}|j}|j} td|jƒD]d} || d||| d|| | d|| <||| || d||| | d| | <q.|j||| S)zQ Additive and Additive Damped Minimization Function (A,) & (Ad,) rrNrOrrrÚ holt_add_damÆs02rQc Csø||j|j t¡<|jr2t|j|j|jƒ\}}}n|jdd…\}}}|jdd…\}}}|jdd…}d|} d|} d|} ||j} ||j} d|jdd…<d|j dd…<d|j dd…<||jd<||j d<||j d|j …<||||| | | | | f S)z3Initialization for the Holt Winters Seasonal ModelsNr)r=rr) rrr>r?rr:rrr r!r"r)r@r<rArBÚgammarDrErFÚs0rGrHÚgammacrIÚy_gammarrrÚ holt_win_initÕs,ÿ     rVc Cs¼t||ƒ\ }}}}}}}}}|j}|j}|j} td|jƒD]d} || d|| d||| d|| <|| d|| d||| d|| | d<q:|j||d| d …S)zD Multiplicative Seasonal Minimization Function (,M) rN©rVr r"rrKrr) r@r<rCrGrTrIrUr r"rrLrrrÚ holt_win__mulïsÿ,6rXc CsÄt||ƒ\ }}}}}}}}}|j} |j} |j} td|jƒD]l} || d|| | d|| | d| | <|| d|| | d|| | d| | | d<q:|j| | d| d …S)z> Additive Seasonal Minimization Function (,A) rNrW) r@r<rArCrRrGrTrIrUr r"rrLrrrÚ holt_win__addsÿ*ÿ*ÿrYc Cst||ƒ\ }}}}}}}}} |j} |j} |j} |j} td|jƒD]´}||d| |d|| |d|| |d| |<|| || |d||| |d| |<| |d| |d|| |d|| |d| || d<q@|j| || | d| d …S)zp Additive and Additive Damped with Multiplicative Seasonal Minimization Function (A,M) & (Ad,M) rN)rVr r!r"rrKrr)r@r<rCrBrFrGrHrTrIrUr r!r"rrLrrrÚholt_win_add_mul_dams0öÿ0&ÿrZc Cst||ƒ\ }}}}}}}}} |j} |j} |j} |j} td|jƒD]´}||d| |d|| |d| |d|| |<|| || |d|| |d|| |<| |d| |d| |d||| |d| || d<q@|j| | || d| d …S)z| Multiplicative and Multiplicative Damped with Multiplicative Seasonal Minimization Function (M,M) & (Md,M) rN©rVr r"r!rrKrr)r@r<rCrBrFrGrHrTrIrUr r"r!rrLrrrÚholt_win_mul_mul_dam7s0öÿ0&ÿr\c Cs"t||ƒ\ }}}}}}}} } |j} |j} |j} |j}td|jƒD]¼}| |d|| |d|| |d|| |d| |<|| || |d||| |d| |<| |d|| |d|| |d|| |d| ||d<q@|j| || | d|d …S)zj Additive and Additive Damped with Additive Seasonal Minimization Function (A,A) & (Ad,A) rNr[©r@r<rArBrRrFrGrHrTrIrUr r"r!rrLrrrÚholt_win_add_add_damWs<ö ÿþÿ0 ÿþÿr^c Cs"t||ƒ\ }}}}}}}} } |j} |j} |j} |j}td|jƒD]¼}| |d|| |d|| |d| |d|| |<|| || |d|| |d|| |<| |d|| |d| |d||| |d| ||d<q@|j| || | d|d …S)zv Multiplicative and Multiplicative Damped with Additive Seasonal Minimization Function (M,A) & (M,Ad) rNr[r]rrrÚholt_win_mul_add_dam{s<ö ÿþÿ0 ÿþÿr_)ZnumpyrÚsqrtZfinfoÚfloatZepsr4rr:r;rJrMrPrQrVrXrYrZr\r^r_rrrrÚs>'$  $