%g6!dZddlmZddlmZmZmZmZddlZ ddl Z ddl m Z ddlmZddlmZdd lmZdd lmZGd d Zy) zCContainer for the results of running autodiff variational inference) OrderedDict)DictOptionalTupleUnionN)Method)scan_variational_csv) get_logger)InferenceMetadata)RunSetc BeZdZdZdeddfdZdefdZdedee je ffdZ d eddfd Z edefd Zedeed ffd Zede fdZede jfdZedej,fdZedeee jffdZedefdZdddedeedee je ffdZdddeedeeee je fffdZede jfdZ edej,fdZ!ddeeddfdZ"y) CmdStanVBzn Container for outputs from CmdStan variational run. Created by :meth:`CmdStanModel.variational`. runsetreturnNc|jtjk(s$tdj |j||_|j |jdy)zInitialize object.zEWrong runset method, expecting variational inference, found method {}rN)methodr VARIATIONAL ValueErrorformatr_set_variational_attrs csv_files)selfrs Q/var/www/dash_apps/app1/venv/lib/python3.12/site-packages/cmdstanpy/stanfit/vb.py__init__zCmdStanVB.__init__sV}} 2 22""(&"7   ##F$4$4Q$78cbdj|jj|jjjj dg}dj|dj |jjdj |jj}|S)NzCmdStanVB: model={}{}r)cmdz#{} csv_file: {} output_file: {}z ) rrmodel_args method_argscomposejoinr stdout_files)rreprs r__repr__zCmdStanVB.__repr__!s&-- KK  t{{00<<DDQBDO ;AA  KK -- . KK 00 1   rattrc|jdrtd| |j|S#t$r}t|jd}~wwxYw)z)Synonymous with ``fit.stan_variable(attr)_zUnknown variable name N) startswithAttributeError stan_variablerargs)rr'es r __getattr__zCmdStanVB.__getattr__-sV ??3  #9$!@A A *%%d+ + * !&&) ) *s2 AA  A sample_csv_0ct|}t||_|d|_|d|_|d|_|d|_y)N column_namesetavariational_meanvariational_sample)r r _metadata _column_names_eta_variational_mean_variational_sample)rr0metas rrz CmdStanVB._set_variational_attrs7sL#L1*40.2>.B; -12D-E/34H/I rc,t|jS)z Total number of information items returned by sampler. Includes approximation information and names of model parameters and computed quantities. )lenr7rs rcolumnszCmdStanVB.columns@s4%%&&r.c|jS)z Names of information items returned by sampler for each draw. Includes approximation information and names of model parameters and computed quantities. )r7r>s rr2zCmdStanVB.column_namesIs!!!rc|jS)z3 Step size scaling parameter 'eta' )r8r>s rr3z CmdStanVB.etaRs yyrc|jS)zB Returns inferred parameter means as numpy array. )r9r>s rvariational_params_npzCmdStanVB.variational_params_npYs %%%rcZtj|jg|jS)zG Returns inferred parameter means as pandas DataFrame. r?)pd DataFramer9r2r>s rvariational_params_pdzCmdStanVB.variational_params_pd`s$ ||T334d>O>OPPrcTtt|j|jS)z)Returns inferred parameter means as Dict.)rzipr2r9r>s rvariational_params_dictz!CmdStanVB.variational_params_dictgs"3t00$2H2HIJJrc|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. )r6r>s rmetadatazCmdStanVB.metadatals~~rmeanvarrOc|tjdd}|r |j}n |j} |jj |j |}|jdk(s|jdk(r+|rtjd|jS|S#t$rDtd|ddj|jj jzwxYw) aw 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, with a leading axis added for the number of draws from the variational approximation. * If the variable is a scalar variable, the return array has shape ( draws, ). * If the variable is a vector, the return array has shape ( draws, len(vector)) * If the variable is a matrix, the return array has shape ( draws, size(dim 1), size(dim 2) ) * If the variable is an array with N dimensions, the return array has shape ( draws, size(dim 1), ..., size(dim N)) This functionaltiy is also available via a shortcut using ``.`` - writing ``fit.a`` is a synonym for ``fit.stan_variable("a")`` :param var: variable name :param mean: if True, return the variational mean. Otherwise, return the variational sample. The default behavior will change in a future release to return the variational sample. See Also -------- CmdStanVB.stan_variables CmdStanMCMC.stan_variable CmdStanMLE.stan_variable CmdStanPathfinder.stan_variable CmdStanGQ.stan_variable CmdStanLaplace.stan_variable zThe default behavior of CmdStanVB.stan_variable() will change in a future release to return the variational sample, rather than the mean. To maintain the current behavior, pass the argument mean=TrueT)r zThe default behavior of CmdStanVB.stan_variable(mean=True) will change in a future release to always return a numpy.ndarray, even for scalar variables.zUnknown variable name: z Available variables are z, ) r warningr9r:r6 stan_varsextract_reshapeshapeitemKeyErrorrr#keys)rrPrOdrawsouts rr,zCmdStanVB.stan_variableusL < L  D **E,,E "nn66s;KKCyyB#))t"3L((0 xxz!J )#/++))DNN4499;<=  sA0B-+B--A C:cli}|jjD]}|j||||<|S)a Return a dictionary mapping Stan program variables names to the corresponding numpy.ndarray containing the inferred values. See Also -------- CmdStanVB.stan_variable CmdStanMCMC.stan_variables CmdStanMLE.stan_variables CmdStanGQ.stan_variables CmdStanPathfinder.stan_variables CmdStanLaplace.stan_variables rN)r6rTr,)rrOresultnames rstan_variableszCmdStanVB.stan_variabless@ NN,, ?D--d->F4L ? rc|jS)z6Returns the set of approximate posterior output draws.)r:r>s rr5zCmdStanVB.variational_samples'''rcXtj|j|jS)zf Returns the set of approximate posterior output draws as a pandas DataFrame. rE)rFrGr:r2r>s rvariational_sample_pdzCmdStanVB.variational_sample_pds! ||D44d>O>OPPrdirc:|jj|y)a Move output CSV files to specified directory. If files were written to the temporary session directory, clean filename. E.g., save 'bernoulli-201912081451-1-5nm6as7u.csv' as 'bernoulli-201912081451-1.csv'. :param dir: directory path See Also -------- stanfit.RunSet.save_csvfiles cmdstanpy.from_csv N)r save_csvfiles)rrcs rrezCmdStanVB.save_csvfiless !!#&r)N)#__name__ __module__ __qualname____doc__r rstrr&rnpndarrayfloatr/rpropertyintr?rr2r3rCrFrGrHrrKr rMrboolr,r_r5rbrerRrrrrs 9v9$9 # **bjj%.?(@*J3J4J'''"eCHo""U &rzz&& Qr||QQ Kc2::o)>KK+37II!)$I rzz5 !IX)-~ c5U*++ ,*(BJJ((Qr||QQ'#'$'rr)ri collectionsrtypingrrrrnumpyrkpandasrFcmdstanpy.cmdstan_argsrcmdstanpy.utilsr cmdstanpy.utils.loggingr rMr rr rrRrrrxs2I#//)0.'_'_'r