"gP&ddlZddlmZddlmZddlmZmZddl m Z ddl m Z dZ dd Zdd Z dd Z dd Z dd Zy)N) arima_process)prefix_dtype_map) _get_epsilonapprox_fprime_cs)find_best_blas_type)_arma_innovationszThe model's autoregressive parameters (ar_params) indicate that the process is non-stationary. The innovations algorithm cannot be used. c Vtj|d}|jdk(}|r |dddf}tj|gn|}tj|gn|}|j\}}tj d| f} tj d|f} |&t |||tj|g\}} } t|} tj|| }tj|| }tj|| }| |j}tt|dz} tt|dz}tt|dz}tj| | || |z }| | | |\}}||||||\}}tj|}tj|d ksFtj |j#r#tj |j#st%t&g}t)|D]8}tj||dd|f|||}|j+|:tj,|j.}|r||dddfd zz}|r|j1}||fS) ay Compute innovations using a given ARMA process. Parameters ---------- endog : ndarray The observed time-series process, may be univariate or multivariate. ar_params : ndarray, optional Autoregressive parameters. ma_params : ndarray, optional Moving average parameters. sigma2 : ndarray, optional The ARMA innovation variance. Default is 1. normalize : bool, optional Whether or not to normalize the returned innovations. Default is False. prefix : str, optional The BLAS prefix associated with the datatype. Default is to find the best datatype based on given input. This argument is typically only used internally. Returns ------- innovations : ndarray Innovations (one-step-ahead prediction errors) for the given `endog` series with predictions based on the given ARMA process. If `normalize=True`, then the returned innovations have been "whitened" by dividing through by the square root of the mean square error. innovations_mse : ndarray Mean square error for the innovations. WC) requirementsrNdtypearma_transformed_acovf_fastarma_innovations_algo_fastarma_innovations_filter)sigma2nobsrg?)nprequirendim atleast_1dshaper_rarrayrasfortranarrayitemgetattrr r arma_acovfanyisfiniteall ValueErrorNON_STATIONARY_ERRORrangeappendvstackTsqueeze)endog ar_params ma_paramsr normalizeprefixsqueezedrk_endogarmar_rrrracovfacovf2thetavuiu_is i/var/www/dash_apps/app1/venv/lib/python3.12/site-packages/statsmodels/tsa/innovations/arma_innovations.pyarma_innovationsr;sB JJuD 1EzzQHag I$5b9EI I$5b9EIKKMD' q9*} B q)| B~. Iy"((6*: ;=q V $E   e5 1E!!)59I!!)59I 6]   !F#*6$AA#C!(6$@@"B%6$==?))"b17dDFLMJ/B CME6)$ 9*/9HE1  A q1u  E"&&( A""$-.. A 7^hh.uQT{I/8%AB   ! A Qq$wZ_ IIK a4KcLt|||||}tj|S)a Compute the log-likelihood of the given data assuming an ARMA process. Parameters ---------- endog : ndarray The observed time-series process. ar_params : ndarray, optional Autoregressive parameters. ma_params : ndarray, optional Moving average parameters. sigma2 : ndarray, optional The ARMA innovation variance. Default is 1. prefix : str, optional The BLAS prefix associated with the datatype. Default is to find the best datatype based on given input. This argument is typically only used internally. Returns ------- float The joint loglikelihood. )r*r+rr-)arma_loglikeobsrsum)r)r*r+rr-llf_obss r: arma_loglikerAns'0eyI%+Fr>s2 HHUOE I$5b9EI I$5b9EI ~. Iy"((6*: ;=q V $E  e 4E!!)59I!!)59I 6]   !F $f/E&E FD y)V 44r<c |gn|}|gn|}t|t|  fd}tj|||f}t|ddt|}t |||S)a Compute the score (gradient of the log-likelihood function). Parameters ---------- endog : ndarray The observed time-series process. ar_params : ndarray, optional Autoregressive coefficients, not including the zero lag. ma_params : ndarray, optional Moving average coefficients, not including the zero lag, where the sign convention assumes the coefficients are part of the lag polynomial on the right-hand-side of the ARMA definition (i.e. they have the same sign from the usual econometrics convention in which the coefficients are on the right-hand-side of the ARMA definition). sigma2 : ndarray, optional The ARMA innovation variance. Default is 1. prefix : str, optional The BLAS prefix associated with the datatype. Default is to find the best datatype based on given input. This argument is typically only used internally. Returns ------- ndarray Score, evaluated at the given parameters. Notes ----- This is a numerical approximation, calculated using first-order complex step differentiation on the `arma_loglike` method. Nc>t|d|z|zdSN)rAparamsr)pqs r:rEzarma_score..funcs.E6"1:vaAq1uvOOr<@lenrrrr r)r*r+rr-rEparams0epsilonrKrLs ` @@r: arma_scorerSspD 'YI'YI IA IAPeeIy&01G7Bc'l;G GT7 33r<c |gn|}|gn|}t|t|  fd}tj|||f}t|ddt|}t |||S)a Compute the score (gradient) per observation. Parameters ---------- endog : ndarray The observed time-series process. ar_params : ndarray, optional Autoregressive coefficients, not including the zero lag. ma_params : ndarray, optional Moving average coefficients, not including the zero lag, where the sign convention assumes the coefficients are part of the lag polynomial on the right-hand-side of the ARMA definition (i.e. they have the same sign from the usual econometrics convention in which the coefficients are on the right-hand-side of the ARMA definition). sigma2 : ndarray, optional The ARMA innovation variance. Default is 1. prefix : str, optional The BLAS prefix associated with the datatype. Default is to find the best datatype based on given input. This argument is typically only used internally. Returns ------- ndarray Score per observation, evaluated at the given parameters. Notes ----- This is a numerical approximation, calculated using first-order complex step differentiation on the `arma_loglike` method. Nc>t|d|z|zdSrH)r>rIs r:rEzarma_scoreobs..funcs3ufRaj&1q5/%a!ef~/ /r<rMrNrPs ` @@r: arma_scoreobsrVsoD 'YI'YI IA IA/eeIy&01G7Bc'l;G GT7 33r<)NNrFN)NNrN)numpyrstatsmodels.tsar statsmodels.tsa.statespace.toolsrstatsmodels.tools.numdiffrrscipy.linalg.blasrr r#r;rAr>rSrVr<r:r^sb)=D1 DE-1\~:CD(5V>?-4`AB.4r<