%g%dZddlmZmZmZmZmZmZmZm Z ddl Z ddl Z ddlZdZddlmZddlmZddlmZd d lmZd d lmZd d lmZGd dZy#e$rdZY8wxYw)zB Container for the result of running a laplace approximation. )AnyDictHashableListMutableMappingOptionalTupleUnionNTF)Method)build_xarray_data)scan_generic_csv)InferenceMetadata) CmdStanMLE)RunSetceZdZdededdfdZddZdedejfdZ de eejffd Z de eejffd Z dejfd Z dd eeeedfdej$fd Z dd eeeedfddfdZedefdZedefdZdefdZdedejfdZdefdZedeedffdZddeeddfdZ y)CmdStanLaplacerunsetmodereturnNc&|jtjk(s$tdj |j||_||_tjd|_ t|jd}t||_ y)zInitialize object.z>Wrong runset method, expecting laplace runset, found method {}rN)methodr LAPLACE ValueErrorformat_runset_modenparray_drawsr csv_filesr _metadata)selfrrconfigs V/var/www/dash_apps/app1/venv/lib/python3.12/site-packages/cmdstanpy/stanfit/laplace.py__init__zCmdStanLaplace.__init__(sr}}.""(&"7   "$((2, !&"2"21"56*62cn|jjdk7ryt|jjdd5}|j j dr! |j j dr!tj|tddd|_dddy#1swYyxYw)N)rrr#,)dtypendmin delimitercomments) r!shapeopenrr"readline startswithrloadtxtfloat)r$fds r&_assemble_drawszCmdStanLaplace._assemble_draws7s ;;   $  $,,((+S 1 R;;=,,S1;;=,,S1** DK   sAB+?#B++B4varc,|j |jj|j|j}|S#t $rDt d|ddj|jjjzwxYw)a\ Return a numpy.ndarray which contains the estimates for the for the named Stan program variable where the dimensions of the numpy.ndarray match the shape of the Stan program variable. This functionaltiy is also available via a shortcut using ``.`` - writing ``fit.a`` is a synonym for ``fit.stan_variable("a")`` :param var: variable name See Also -------- CmdStanMLE.stan_variables CmdStanMCMC.stan_variable CmdStanPathfinder.stan_variable CmdStanVB.stan_variable CmdStanGQ.stan_variable zUnknown variable name: z Available variables are z, ) r9r# stan_varsextract_reshaper!KeyErrorrjoinkeys)r$r:outs r& stan_variablezCmdStanLaplace.stan_variableFs&  "nn66s;KK CJ )#/++))DNN4499;<=  s 3AA Bchi}|jjD]}|j|||<|S)a Return a dictionary mapping Stan program variables names to the corresponding numpy.ndarray containing the inferred values. :param inc_warmup: When ``True`` and the warmup draws are present in the MCMC sample, then the warmup draws are included. Default value is ``False`` See Also -------- CmdStanGQ.stan_variable CmdStanMCMC.stan_variables CmdStanMLE.stan_variables CmdStanPathfinder.stan_variables CmdStanVB.stan_variables )r#r<rB)r$resultnames r&stan_variableszCmdStanLaplace.stan_variablesgs;"NN,, 4D--d3F4L 4 r(c|j|jjjDcic]!\}}||j |j #c}}Scc}}w)aH Returns a dictionary of all sampler variables, i.e., all output column names ending in `__`. Assumes that all variables are scalar variables where column name is variable name. Maps each column name to a numpy.ndarray (draws x chains x 1) containing per-draw diagnostic values. )r9r# method_varsitemsr=r!)r$rEr:s r&method_variableszCmdStanLaplace.method_variables}s] "^^77==? c #%%dkk2 2   s&A"c:|j|jS)z Return a numpy.ndarray containing the draws from the approximate posterior distribution. This is a 2-D array of shape (draws, parameters). )r9r!r$s r&drawszCmdStanLaplace.drawss {{r(varscL|t|tr|g}n|}|jg}|tj D]}||j j vr|j|-||j jvrL|j j|}|j|j|j|jtd|nt|j}tj |j"|j|S)NzUnknown variable: )columns) isinstancestrr9dictfromkeysr#rHappendr<extend column_names start_idxend_idxrlistpd DataFramer!)r$rN vars_listcolsr:infos r&draws_pdzCmdStanLaplace.draws_pds  $$!F     }}Y/ A$..444KK$DNN444>>33C8DKK))$..4<<H%'9#%?@@ A))*D||DKK1B1BCDIIr(z xr.Datasetc ts td|.t|jjj }nt |tr|g}n|}|j|jj}|dd|dd|d|dd}i}d tj|jjd i}|D]H}t||jj||jddtjddfJt!j"||| j%d d j'S) z Returns the sampler draws as a xarray Dataset. :param vars: optional list of variable names. See Also -------- CmdStanMCMC.draws_xr CmdStanGQ.draws_xr z>Package "xarray" is not installed, cannot produce draws array.Nstan_version_major.stan_version_minorstan_version_patchmodel) stan_versionrfdrawr)coordsattrs.)XARRAY_INSTALLED RuntimeErrorrZr#r<r@rQrRr9cmdstan_configraranger!r2r newaxisxrDataset transposesqueeze)r$rNr]metarjdata coordinatesr:s r&draws_xrzCmdStanLaplace.draws_xrsA P  <T^^55::<=I c "II ~~,,#$89:!()*!D1E,F+GI']0  /1 BIIdkk//236  C ((- Arzz1,-   JJtKu = Yvs # WY r(c|jS)zj Return the maximum a posteriori estimate (mode) as a :class:`CmdStanMLE` object. )rrLs r&rzCmdStanLaplace.modes zzr(c|jS)z Returns object which contains CmdStan configuration as well as information about the names and structure of the inference method and model output variables. )r#rLs r&metadatazCmdStanLaplace.metadatas~~r(cdjt|jjDcgc]}d|z c}dd}dj |j j ||j jjjdg}dj |dj|j jdj|j j}|Scc}w) N  rz&CmdStanLaplace: model={} mode=({}) {}r)cmdz%{} csv_files: {} output_files: {}z ) r?reprr splitlinesrrrf_args method_argscomposer" stdout_files)r$linerreps r&__repr__zCmdStanLaplace.__repr__syy%)$))_%?%?%A BTTD[ B "9?? LL    LL   * * 2 21" 2 =  <BB  KK .. / KK 11 2    Cs C8attrc|jdrtd| |j|S#t$r}t|jd}~wwxYw)z)Synonymous with ``fit.stan_variable(attr)_zUnknown variable name N)r5AttributeErrorrBrargs)r$res r& __getattr__zCmdStanLaplace.__getattr__sV ??3  #9$!@A A *%%d+ + * !&&) ) *s2 AA  Ac:|j|jSN)r9__dict__rLs r& __getstate__zCmdStanLaplace.__getstate__s }}r(.c4|jjdS)a9 Names of all outputs from the sampler, comprising sampler parameters and all components of all model parameters, transformed parameters, and quantities of interest. Corresponds to Stan CSV file header row, with names munged to array notation, e.g. `beta[1]` not `beta.1`. rW)r#rmrLs r&rWzCmdStanLaplace.column_namess~~,,^<-12 CcD()2  2 hj+# ** *d=eCHo==(#($(r(r)__doc__typingrrrrrrr r numpyrpandasr[xarrayrprk ImportErrorcmdstanpy.cmdstan_argsr cmdstanpy.utils.data_mungingr cmdstanpy.utils.stancsvr rzrmlerrrrrr(r&rsf   *:4'I(I(!sAAA