%g(dZddlmZddlmZmZmZmZddlZ ddl Z ddl m Z mZddlmZmZddlmZdd lmZGd d Zy) z0Container for the result of running optimization) OrderedDict)DictOptionalTupleUnionN)Method OptimizeArgs) get_loggerscan_optimize_csv)InferenceMetadata)RunSetc eZdZdZdeddfdZdefdZdedee je ffdZ d eddfd Z edeed ffd Zedefd Zede jfdZedee jfdZedej.fdZedeej.fdZedeee j6ffdZddddedededee je ffdZ ddedeeee je fffdZddeeddfdZ y) CmdStanMLEzh Container for outputs from CmdStan optimization. Created by :meth:`CmdStanModel.optimize`. runsetreturnNc|jtjk(s$tdj |j||_|j |_|j jj}t|tsJ|j|_ |j|jdy)zInitialize object.z?Wrong runset method, expecting optimize runset, found method {}rN)methodrOPTIMIZE ValueErrorformatr_check_retcodes converged_args method_args isinstancer save_iterations_save_iterations_set_mle_attrs csv_files)selfr optimize_argss R/var/www/dash_apps/app1/venv/lib/python3.12/site-packages/cmdstanpy/stanfit/mle.py__init__zCmdStanMLE.__init__s}}/""(&"7  //1 ))55  <   '4&C&C F,,Q/0cdj|jj|jjjj dg}dj|dj |jjdj |jj}|js"dj|}dj|}|S)NzCmdStanMLE: model={}{}r)cmdz#{} csv_file: {} output_file: {}z z{} Warning: invalid estimate, z#{} optimization failed to converge.) rrmodelrrcomposejoinr stdout_filesr)r!reprs r#__repr__zCmdStanMLE.__repr__'s'.. KK  t{{00<<DDQBDO ;AA  KK -- . KK 00 1  ~~5<.B $U   *.{*;DO !r%.c|jS)z Names of estimated quantities, includes joint log probability, and all parameters, transformed parameters, and generated quantities. )r=r!s r#r9zCmdStanMLE.column_namesGs !!!r%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<rBs r#metadatazCmdStanMLE.metadataOs~~r%cd|jstjd|jS)z Returns all final estimates from the optimizer as a numpy.ndarray which contains all optimizer outputs, i.e., the value for `lp__` as well as all Stan program variables. 2Invalid estimate, optimization failed to converge.)rr warningr>rBs r#optimized_params_npzCmdStanMLE.optimized_params_npXs*~~ L D yyr%c|jstjdy|jstjd|jS)z Returns all saved iterations from the optimizer and final estimate as a numpy.ndarray which contains all optimizer outputs, i.e., the value for `lp__` as well as all Stan program variables. lIntermediate iterations not saved to CSV output file. Rerun the optimize method with "save_iterations=True".NrF)rr rGrr?rBs r#optimized_iterations_npz"CmdStanMLE.optimized_iterations_npesN$$ L I ~~ L D r%c|jjstjdt j |j g|jS)z Returns all final estimates from the optimizer as a pandas.DataFrame which contains all optimizer outputs, i.e., the value for `lp__` as well as all Stan program variables. rFcolumns)rrr rGpd DataFramer>r9rBs r#optimized_params_pdzCmdStanMLE.optimized_params_pdysF{{**, L D ||TYYK1B1BCCr%c|jstjdy|jstjdt j |j |jS)z Returns all saved iterations from the optimizer and final estimate as a pandas.DataFrame which contains all optimizer outputs, i.e., the value for `lp__` as well as all Stan program variables. rJNrFrM)rr rGrrOrPr?r9rBs r#optimized_iterations_pdz"CmdStanMLE.optimized_iterations_pds_$$ L I ~~ L D ||DOOT5F5FGGr%c|jjstjdt t |j |jS)z Returns all estimates from the optimizer, including `lp__` as a Python Dict. Only returns estimate from final iteration. rF)rrr rGrzipr9r>rBs r#optimized_params_dictz CmdStanMLE.optimized_params_dictsE {{**, L D 3t00$))<==r%FTinc_iterationswarnvarrXrYc*||jjvr5td|ddj|jjz|r'|r%|jst j d|r3|jjst j d|r|jr |j}n |j} |jj|j|}|jdk(s|jdk(r)t j d|jS|S#t$rDtd|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 :param inc_iterations: When ``True`` and the intermediate estimates are included in the output, i.e., the optimizer was run with ``save_iterations=True``, then intermediate estimates are included. Default value is ``False``. See Also -------- CmdStanMLE.stan_variables CmdStanMCMC.stan_variable CmdStanPathfinder.stan_variable CmdStanVB.stan_variable CmdStanGQ.stan_variable CmdStanLaplace.stan_variable zUnknown variable name: z Available variables are z, rJrF)r zThe default behavior of CmdStanMLE.stan_variable() will change in a future release to always return a numpy.ndarray, even for scalar variables.)r< stan_varsrr*rr rGrrr?r>extract_reshapeshapeitemKeyErrorkeys)r!rZrXrYdataouts r#r3zCmdStanMLE.stan_variablest> dnn.. .)#/++-1YYt~~7O7O-PQ  N4+@+@ L I   335 L D  d33??D99D "nn66s;KKCyyB#))t"3 $$@ xxz!J )#/++))DNN4499;<=  sA.EEA Fc|jjstjdi}|jj D]}|j ||d||<|S)a Return a dictionary mapping Stan program variables names to the corresponding numpy.ndarray containing the inferred values. :param inc_iterations: When ``True`` and the intermediate estimates are included in the output, i.e., the optimizer was run with ``save_iterations=True``, then intermediate estimates are included. Default value is ``False``. See Also -------- CmdStanMLE.stan_variable CmdStanMCMC.stan_variables CmdStanPathfinder.stan_variables CmdStanVB.stan_variables CmdStanGQ.stan_variables CmdStanLaplace.stan_variables rFFrW)rrr rGr<r]r3)r!rXresultnames r#stan_variableszCmdStanMLE.stan_variablessn,{{**, L D NN,, D--^%.F4L  r%dirc:|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)r!ris r#rkzCmdStanMLE.save_csvfiless !!#&r%)F)N)!__name__ __module__ __qualname____doc__rr$strr-rnpndarrayfloatr6rpropertyrr9r rDrHrrKrOrPrQrSrfloat64rVboolr3rhrkr\r%r#rrs 1v1$1" # **bjj%.?(@*<3<4<"eCHo""+ RZZ  "**)=& DR\\ D DH",,)?HH& >tCO'< > > % E E E  E rzz5 ! EP&+" c5U*++ ,B'#'$'r%r)ro collectionsrtypingrrrrnumpyrqpandasrOcmdstanpy.cmdstan_argsrr cmdstanpy.utilsr r rDr rrrr\r%r#r}s/6#//79'L'L'r%