%g pdZddlmZmZmZmZmZddlZddl m Z ddl m Z ddl mZddlmZGdd Zy) z1 Container for the result of running Pathfinder. )DictListOptionalTupleUnionN)Method)InferenceMetadata)RunSet)scan_generic_csvc eZdZdZdefdZ ddeededee e e e jfe e e jfffdZde fd Zdd Zd e de jfd Zde e e jffd Zde e e jffdZde jfdZde de jfdZdefdZedefdZedee dffdZedefdZddee ddfdZy)CmdStanPathfinderzp Container for outputs from the Pathfinder algorithm. Created by :meth:`CmdStanModel.pathfinder()`. runsetc|jtjk(s$tdj |j||_t jd|_t|jd}t||_ y)zInitialize object.zAWrong runset method, expecting Pathfinder runset, found method {}rN) methodr PATHFINDER ValueErrorformat_runsetnparray_drawsr csv_filesr _metadata)selfrconfigs Y/var/www/dash_apps/app1/venv/lib/python3.12/site-packages/cmdstanpy/stanfit/pathfinder.py__init__zCmdStanPathfinder.__init__sm}} 1 11""(&"7  "$((2, !&"2"21"56*62Nseedchainsreturncd|jtjj|}|j |j j d|d}|dk(rY|j |d}|jjjDcic]\}}||j|c}}S|Dcgc]V}|jjjDcic]$\}}||j|j |&c}}Xc}}}Scc}}wcc}}wcc}}}w)a& Create initial values for the parameters of the model by randomly selecting draws from the Pathfinder approximation. :param seed: Used for random selection, defaults to None :param chains: Number of initial values to return, defaults to 4 :return: The initial values for the parameters of the model. If ``chains`` is 1, a dictionary is returned, otherwise a list of dictionaries is returned, in the format expected for the ``inits`` argument. of :meth:`CmdStanModel.sample`. rF)sizereplace) _assemble_drawsrrandom default_rngchoicershaper stan_varsitemsextract_reshape) rr r!rngidxsdrawnamevaridxs r create_initszCmdStanPathfinder.create_inits#s ii##D)zz$++++A.VUzK Q;;;tAw'D"&!9!9!?!?!AD#c))$//   &*^^%=%=%C%C%E!c#--dkk#.>??   sD?,D++)D%D+%D+cbdj|jj|jjjj dg}dj|dj |jjdj |jj}|S)NzCmdStanPathfinder: model={}{}r)cmdz%{} csv_files: {} output_files: {}z ) rrmodel_args method_argscomposejoinr stdout_files)rreps r__repr__zCmdStanPathfinder.__repr__Ds-44 LL   LL   * * 2 21" 2 = <BB  KK .. / KK 11 2   rcn|jjdk7ryt|jjdd5}|j j dr! |j j dr!tj|tddd|_dddy#1swYyxYw)N)rrr#,)dtypendmin delimitercomments) rr+openrrreadline startswithrloadtxtfloat)rfds rr'z!CmdStanPathfinder._assemble_drawsQs ;;   $  $,,((+S 1 R;;=,,S1;;=,,S1** DK   sAB+?#B++B4r3c,|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 -------- CmdStanPathfinder.stan_variables CmdStanMLE.stan_variable CmdStanMCMC.stan_variable CmdStanVB.stan_variable CmdStanGQ.stan_variable CmdStanLaplace.stan_variable zUnknown variable name: z Available variables are z, ) r'rr,r.rKeyErrorrr<keys)rr3outs r stan_variablezCmdStanPathfinder.stan_variable`s(  "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. See Also -------- CmdStanPathfinder.stan_variable CmdStanMCMC.stan_variables CmdStanMLE.stan_variables CmdStanVB.stan_variables CmdStanGQ.stan_variables CmdStanLaplace.stan_variables )rr,rS)rresultr2s rstan_variablesz CmdStanPathfinder.stan_variabless;NN,, 4D--d3F4L 4 rc|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. )r'r method_varsr-r.r)rr2r3s rmethod_variablesz"CmdStanPathfinder.method_variabless] "^^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). )r'rrs rdrawszCmdStanPathfinder.drawss {{rattrc|jdrtd| |j|S#t$r}t|jd}~wwxYw)z)Synonymous with ``fit.stan_variable(attr)_zUnknown variable name N)rKAttributeErrorrSrargs)rr]es r __getattr__zCmdStanPathfinder.__getattr__sV ??3  #9$!@A A *%%d+ + * !&&) ) *s2 AA  Ac:|j|jSN)r'__dict__r[s r __getstate__zCmdStanPathfinder.__getstate__s }}rc|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. )rr[s rmetadatazCmdStanPathfinder.metadatas~~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`. column_names)rcmdstan_configr[s rrkzCmdStanPathfinder.column_namess~~,,^<rs-65)8+4^(^(r