M/PhwLdZddlZddlmZddlZGddZ ddZdS) zM Created on Fri Dec 19 11:29:18 2014 Author: Josef Perktold License: BSD-3 N)statsceZdZdZ d dZedZedZedZedZ edZ dd Z dd Z dS)PredictionResultsaE Results class for predictions. Parameters ---------- predicted_mean : ndarray The array containing the prediction means. var_pred_mean : ndarray The array of the variance of the prediction means. var_resid : ndarray The array of residual variances. df : int The degree of freedom used if dist is 't'. dist : {'norm', 't', object} Either a string for the normal or t distribution or another object that exposes a `ppf` method. row_labels : list[str] Row labels used in summary frame. Nc||_||_||_||_||_||dkrt j|_d|_dS|dkr t j |_|jf|_dS||_d|_dS)Nnormt) predictedvar_preddf var_resid row_labelsrrdist dist_argsr )selfpredicted_mean var_pred_meanr r rrs b/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/statsmodels/regression/_prediction.py__init__zPredictionResults.__init__$sy'% "$ <46>> DIDNNN S[[DI"gZDNNNDIDNNNcDtj|j|jzSN)npsqrtrr rs rse_obszPredictionResults.se_obs6swt)DN:;;;rc|jSr)sers rse_meanzPredictionResults.se_mean:s wrc|jSr)r rs rrz PredictionResults.predicted_mean>s ~rc|jSr)r rs rrzPredictionResults.var_pred_meanCs }rc4tj|jSr)rrrrs rrzPredictionResults.seHswt)***rF皙?c|r|jn|j}|jjd|dz z g|jR}|j||zz }|j||zz}t j||fS)a. Returns the confidence interval of the value, `effect` of the constraint. This is currently only available for t and z tests. Parameters ---------- alpha : float, optional The significance level for the confidence interval. ie., The default `alpha` = .05 returns a 95% confidence interval. Returns ------- ci : ndarray, (k_constraints, 2) The array has the lower and the upper limit of the confidence interval in the columns. g@)rrrppfrrr column_stack)robsalpharqloweruppers rconf_intzPredictionResults.conf_intMsq( 1T[[T\ DIM!ebj. :4> : : :#a"f,#a"f,u~...rcp||d}||d}i}|j|d<|j|d<|dddf|d<|dddf|d <|dddf|d <|dddf|d <||_t j||j| }|S) NT)r)r(Fmeanmean_ser mean_ci_lowerr% mean_ci_upper obs_ci_lower obs_ci_upper)indexcolumns)r-rrtablepd DataFramerkeys)rr)ci_obsci_mean to_includeress r summary_framezPredictionResults.summary_framehsU55--e-77 !0 6 $ 9&-aaadm ?#&-aaadm ?#%+AAAqD\ >"%+AAAqD\ >" l:T_#-??#4#4666 r)NNN)Fr#)r#) __name__ __module__ __qualname____doc__rpropertyrrrrrr-r?rrrrrs*15    $<<X<XXX++X+////6rrTc~|rft|jdrQ|Oddlm}t |t jrt j|}||jjj |}||"t|dd}t|rd}tj |}|jdkr/|jjddkr |dddf}n |dddf}tj|}nA|jj}|t|jdd}|t|jjdd}|Utj |}|jdkr6|jdks|jd|jdkrt)d |i}|jj|j|fi|}|}|tj||jjzd} |j} |jd kr |jd } || |z} d d g|j} t=|| | |j| |S)a Compute prediction results. Parameters ---------- exog : array_like, optional The values for which you want to predict. transform : bool, optional If the model was fit via a formula, do you want to pass exog through the formula. Default is True. E.g., if you fit a model y ~ log(x1) + log(x2), and transform is True, then you can pass a data structure that contains x1 and x2 in their original form. Otherwise, you'd need to log the data first. weights : array_like, optional Weights interpreted as in WLS, used for the variance of the predicted residual. row_labels : list A list of row labels to use. If not provided, read `exog` is available. **kwargs Some models can take additional keyword arguments, see the predict method of the model for the details. Returns ------- linear_model.PredictionResults The prediction results instance contains prediction and prediction variance and can on demand calculate confidence intervals and summary tables for the prediction of the mean and of new observations. formulaNr)dmatrixr5r%weightsrzweights has wrong shapez fixed scalescalerr )r rr) hasattrmodelpatsyrG isinstancer8Seriesr9data design_infogetattrcallablerasarrayndimparamsshape atleast_2dexogsize ValueErrorpredict cov_paramsdotTsumrIcov_typecov_kwdsuse_trdf_resid) rrX transformrHr pred_kwdsrGrcovbrr rs rget_predictionrg}sgF:WTZ33:8H!!!!!! dBI & & &<%%Dwtz2D99    w55J ## "! z$ 9>>{ #a''D!!!G}AAAtG}}T""z ?dj)T::G   ,EEJ*W%% L1  ""gmA&6$*Q-&G&G677 7 'TZ' TGGYGGN ??  DBF4002277::M I } %%M'* W C= $D ^]I $ D(2 4 4 44r)NTNNN) rCnumpyrscipyrpandasr8rrgrrrrkskkkkkkkk\=A.2Z4Z4Z4Z4Z4Z4r