B zdJ%@sddlmZmZmZmZddgZddlZddlZddlZ ddl m Z ddl m Z mZmZdddZejd d d Gd d d eZejd d d Gd ddeZdS))divisionprint_functionabsolute_importunicode_literals make_ladderSamplerN) RandomState)utilchainensemblecfCs"t|tks|dkrtd|dkr4|dkr4td|dk rL|dkrLtd|dk rpt|tksh|dkrptdtddd d d d d ddddddddddddddddddd d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5d6d7d8d9d:d;dd?d@dAdBdCdDdEdFdGdHdIdJdKdLdMdNdOdPdQdRdSdTdUdVdWdXdYdZd[d\d]d^d_d`dadbdcdddedfdgdhdidjgd}||jdkkrvdldmttdnt|}n ||d}do}|tjkrdp}d}|d}|dk r|dkr||d}n.|dkrtdqtt|t|dr}t dkt | |}|rt |dkgf}|S)sa  Returns a ladder of :math:`\beta \equiv 1/T` under a geometric spacing that is determined by the arguments ``ntemps`` and ``Tmax``. The temperature selection algorithm works as follows: Ideally, ``Tmax`` should be specified such that the tempered posterior looks like the prior at this temperature. If using adaptive parallel tempering, per `arXiv:1501.05823 `_, choosing ``Tmax = inf`` is a safe bet, so long as ``ntemps`` is also specified. :param ndim: The number of dimensions in the parameter space. :param ntemps: (optional) If set, the number of temperatures to generate. :param Tmax: (optional) If set, the maximum temperature for the ladder. Temperatures are chosen according to the following algorithm: * If neither ``ntemps`` nor ``Tmax`` is specified, raise an exception (insufficient information). * If ``ntemps`` is specified but not ``Tmax``, return a ladder spaced so that a Gaussian posterior would have a 25% temperature swap acceptance ratio. * If ``Tmax`` is specified but not ``ntemps``: * If ``Tmax = inf``, raise an exception (insufficient information). * Else, space chains geometrically as above (for 25% acceptance) until ``Tmax`` is reached. * If ``Tmax`` and ``ntemps`` are specified: * If ``Tmax = inf``, place one chain at ``inf`` and ``ntemps-1`` in a 25% geometric spacing. * Else, use the unique geometric spacing defined by ``ntemps`` and ``Tmax``. r z'Invalid number of dimensions specified.Nz,Must specify one of ``ntemps`` and ``Tmax``.z ``Tmax`` must be greater than 1.z)Invalid number of temperatures specified.ggj+F9@g@gȓk@g|a2U0 @g:#J{/@g @g!@gԕ@gBs@gm@gY/rb@gEb?g£#?g>"D?gbg?gpwn?g 3mJ?g46rCrHr rUr rrrrrs2      )NN) __future__rrrr__all__r1 itertoolsnumpyrZnumpy.random.mtrandrr r r rsrFrrrrrrs  f