a аму@sЬdZddlmZddlmZddlmZddlmZddl m Z mZddlm Z ddlmZdd lmZdd lmZddlmZmZddlmZdd lmZmZddlmZmZmZddl m!Z!m"Z"ddl#m$Z$m%Z%ddl&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.ddl/m0Z0ddl1m2Z2ddl3m4Z4GddДde'ГZ5GddДdГZ6GddДdГZ7GddДdГZ8e6e8e7dЬZ9GddДde)e(ГZ:Gd d!Дd!e+ГZ;Gd"d#Дd#e)ГZt|tГrt||ГSt|tГr(t||ГSt|Г}tа|||бSйN)┌ isinstancerrrrr r┌__new__)┌cls┌sym┌distйr+·d/Users/vegardjervell/Documents/master/model/venv/lib/python3.9/site-packages/sympy/stats/joint_rv.pyr'*s zJointPSpace.__new__cCs|jjSr%)┌domain┌setй┌selfr+r+r,r.2szJointPSpace.setcCs |jdS)Nrй┌argsr/r+r+r,┌symbol6szJointPSpace.symbolcCs |jdSйNщr1r/r+r+r,┌distribution:szJointPSpace.distributioncCst|j|ГSr%)┌JointRandomSymbolr3r/r+r+r,┌value>szJointPSpace.valuecCs>|jj}t|tГr tt|jГГSt|tГr8|jddStj S)Nrщ ) r6r.r&r r┌lenr2r┌limitsZOne)r0┌_setr+r+r,┌component_countBs zJointPSpace.component_countcs"ЗfddДtИjГDГ}Иj|ОS)Ncsg|]}tИj|ГСqSr+йrr3й┌.0┌ir/r+r,┌ Mєz#JointPSpace.pdf..)┌ranger=r6йr0r)r+r/r,┌pdfKszJointPSpace.pdfcCs0t|jГ}|st|j|jjГStddД|DГОS)NcSsg|]}|jjСqSr+)┌pspacer-йr@┌rvr+r+r,rBUrCz&JointPSpace.domain..)rr6rr3r.rйr0┌rvsr+r+r,r-Ps zJointPSpace.domaincCs|jj|Sr%)r.r2)r0┌indexr+r+r,┌component_domainWszJointPSpace.component_domainc sИj}|аtбrtdГВЗfddДt|ГDГ}ddД|DГ}tt||ГГ}tЗfddД|DГГЙtЗfddД|DГГ}d}t|ГD]<}||vrВ||а Иj jj|бt||Г||<|d 7}qВИj j rцtИtИj |Оg|вRОГ} n&Иj jРrtИtИj |Оg|вRОГ} | а|бS) Nz_Marginal distributions cannot be computed for symbolic dimensions. It is a work under progress.csg|]}tИj|ГСqSr+r>r?r/r+r,rB_rCz5JointPSpace.marginal_distribution..cSsg|]}tt|ГГСqSr+)r┌strr?r+r+r,rB`rCc3s"|]}tttИj|ГГГVqdSr%)rrNrr3r?r/r+r,┌ brCz4JointPSpace.marginal_distribution..c3s|]}|Иvr|gVqdSr%r+r?)r)r+r,rOcrCrr5)r=Zatomsr┌ ValueErrorrD┌dict┌zip┌tuple┌list┌appendr6r.r2┌ is_Continuousrr┌is_Discreter┌xreplace) r0┌indices┌count┌origZall_symsZreplace_dictr;rLrA┌fr+rEr,┌marginal_distributionZs& z!JointPSpace.marginal_distributionNFcs╪tЗfddДtИjГDГГ}Иp"|ЙtЗfddД|DГГs>|S|Иj}ИD]J}t|tГr||а|tt|j Г|j dГiб}qLt|tГrL|а||jiб}qLИj t|Гvr▓ttdГГВtЗfddД|DГГ}t|g|вRОS)Nc3s|]}Иj|VqdSr%)r8r?r/r+r,rOqrCz2JointPSpace.compute_expectation..c3s|]}|ИvVqdSr%r+r?йrKr+r,rOsrCr5zq Expectations of expression with unindexed joint random symbols cannot be calculated yet.c3s8|]0}tt|jГ|jdГИjjj|jdfVqdS)r5N)rrN┌baser2r6r.rHr/r+r,rOs )rSrDr=┌anyrFr&rrXrNr_r2rr3r8r┌NotImplementedErrorr"r)r0┌exprrK┌evaluate┌kwargs┌symsrIr;r+)rKr0r,┌compute_expectationps " zJointPSpace.compute_expectationcCs tГВdSr%йraйr0┌ conditionr+r+r,┌whereГszJointPSpace.wherecCs tГВdSr%rg)r0rbr+r+r,┌compute_densityЖszJointPSpace.compute_densityr+┌scipycCst|j|Г|jj|||dНiS)zo Internal sample method Returns dictionary mapping RandomSymbol to realization value. )┌library┌seed)rr3r6┌sample)r0┌sizermrnr+r+r,roЙs zJointPSpace.samplecCs tГВdSr%rgrhr+r+r,┌probabilityТszJointPSpace.probability)NF)r+rlN)┌__name__┌ __module__┌__qualname__┌__doc__r'┌propertyr.r3r6r8r=rFr-rMr]rfrjrkrorqr+r+r+r,r$%s. r$c@s&eZdZdZdddДZeddДГZdS)┌SampleJointScipyz7Returns the sample from scipy of the given distributionNcCs|а|||бSr%)┌ _sample_scipyйr(r*rprnr+r+r,r'ШszSampleJointScipy.__new__c s└ddl}|dust|tГr*|jj|dНЙn|ЙddlmЙЗЗfddДЗЗfddДЗЗfddДd Ь}d dДddДddДd Ь}|аб}|jj |vrФdS||jj ||Г}|а |||jj |ГбS) zSample from SciPy.rNйrn)┌statscs$Иjjt|jГабt|jГ|ИdНS)N)┌mean┌covrp┌random_state)┌multivariate_normalrKr┌mu┌flatten┌sigmaйr*rpй┌ rand_stateZscipy_statsr+r,┌жs■z0SampleJointScipy._sample_scipy..csИjjt|jtГаб|ИdНS)N)┌alpharpr~)┌ dirichletrKrrЗ┌floatrБrГrДr+r,rЖйs cs&Иjjt|jГt|jtГаб|ИdНS)N)┌n┌prpr~)┌multinomialrK┌intrКrrЛrЙrБrГrДr+r,rЖлs йZMultivariateNormalDistributionZMultivariateBetaDistributionZMultinomialDistributioncSst|jГабjSr%йrrАrБ┌shapeйr*r+r+r,rЖ░rCcSst|jГабjSr%йrrЗrБrРrСr+r+r,rЖ▒rCcSst|jГабjSr%йrrЛrБrРrСr+r+r,rЖ▓rC)┌numpyr&rН┌random┌default_rngrlr{┌keys┌ __class__rr┌reshape) r(r*rprnrФZscipy_rv_map┌sample_shape┌ dist_list┌samplesr+rДr,rxЫs$·¤zSampleJointScipy._sample_scipy)N)rrrsrtrur'┌classmethodrxr+r+r+r,rwЦs rwc@s&eZdZdZdddДZeddДГZdS)┌SampleJointNumpyz7Returns the sample from numpy of the given distributionNcCs|а|||бSr%)┌ _sample_numpyryr+r+r,r'└szSampleJointNumpy.__new__c s▓ddl}|dust|tГr*|jj|dНЙn|ЙЗfddДЗfddДЗfddДdЬ}d dДd dДddДdЬ}|аб}|jj|vrВdS||jj|t|ГГ}|а |||jj|ГбS)zSample from NumPy.rNrzcs$Иjt|jtГабt|jtГ|dНS)N)r|r}rp)rrrАrЙrБrВrГйrЕr+r,rЖ═s■z0SampleJointNumpy._sample_numpy..csИjt|jtГаб|dНS)N)rЗrp)rИrrЗrЙrБrГrаr+r,rЖ╨s cs"Иjt|jГt|jtГаб|dНS)N)rКZpvalsrp)rМrНrКrrЛrЙrБrГrаr+r,rЖ╥s rОcSst|jГабjSr%rПrСr+r+r,rЖ╫rCcSst|jГабjSr%rТrСr+r+r,rЖ╪rCcSst|jГабjSr%rУrСr+r+r,rЖ┘rC) rФr&rНrХrЦrЧrШrrrrЩ) r(r*rprnrФZnumpy_rv_maprЪrЫrЬr+rаr,rЯ├s" ·¤zSampleJointNumpy._sample_numpy)N)rrrsrtrur'rЭrЯr+r+r+r,rЮ╜s rЮc@s&eZdZdZdddДZeddДГZdS)┌SampleJointPymcz7Returns the sample from pymc3 of the given distributionNcCs|а|||бSr%)┌ _sample_pymc3ryr+r+r,r'чszSampleJointPymc.__new__c sюddlЙЗfddДЗfddДЗfddДdЬ}ddДd dДd dДdЬ}|аб}|jj|vr\dSddl}|аdба|jбИабПF||jj|ГИj t |Гdd |d d dНddЕd}WdГn1s╩0Y|а|||jj|ГбS)zSample from PyMC3.rNcs2Иjdt|jtГабt|jtГd|jjdfdНS)N┌Xr5r)rАr}rР)ZMvNormalrrАrЙrБrВrРrСй┌pymc3r+r,rЖЁs z/SampleJointPymc._sample_pymc3..csИjdt|jtГабdНS)Nrг)┌a)Z DirichletrrЗrЙrБrСrдr+r,rЖєscs.Иjdt|jГt|jtГабdt|jГfdНS)Nrгr5)rКrЛrР)ZMultinomialrНrКrrЛrЙrБr:rСrдr+r,rЖїs rОcSst|jГабjSr%rПrСr+r+r,rЖ√rCcSst|jГабjSr%rТrСr+r+r,rЖ№rCcSst|jГабjSr%rУrСr+r+r,rЖ¤rCrеr5F)ZdrawsZchainsZprogressbarZrandom_seedZreturn_inferencedataZcompute_convergence_checksrг)rеrЧrШrr┌logging┌ getLogger┌setLevel┌ERRORZModelrorrЩ) r(r*rprnZpymc3_rv_maprЪrЫrзrЬr+rдr,rвъs$ ·¤ DzSampleJointPymc._sample_pymc3)N)rrrsrtrur'rЭrвr+r+r+r,rбфs rб)rlrеrФc@sNeZdZdZdZddДZeddДГZeddДГZd d ДZ dddДZ ddДZd S)┌JointDistributionzУ Represented by the random variables part of the joint distribution. Contains methods for PDF, CDF, sampling, marginal densities, etc. йrFcGsPttt|ГГ}tt|ГГD]"}t||tГrt||Г||<qtj|g|вRОSr%) rT┌maprrDr:r&rrr')r(r2rAr+r+r,r's zJointDistribution.__new__cCs t|jГSr%)r┌symbolsr/r+r+r,r-"szJointDistribution.domaincCs|jjdSr4)┌densityr2r/r+r+r,rF&szJointDistribution.pdfcCs║t|tГstd|t|ГfГВ|аб}|jjj}|аt ddД|j DГГб}tt|ГГD]^}||j rИt|||||j|||fГ}qV||jrVt|||||j|||fГ}qV|S)Nz!%s should be of type dict, got %scss|]}|jdVqdSйrNr1r?r+r+r,rO/rCz(JointDistribution.cdf..)r&rQrP┌typerЧr-r.ZsetsrFrSrоrDr:rVr┌infrWr )r0┌otherrKr<rbrArпr+r+r,┌cdf*s zJointDistribution.cdfr+rlNcCsjgdв}||vr tdt|ГГВt|Гs4td|ГВt||||dН}|durR|Std|jj|fГВdS)z, A random realization from the distribution )rlrФrеz&Sampling from %s is not supported yet.zFailed to import %srzNz4Sampling for %s is not currently implemented from %s)rarNr#rP┌_get_sample_class_jrvrШrr)r0rprmrn┌ librariesZsampsr+r+r,ro9s zJointDistribution.samplecGs |j|ОSr%rмйr0r2r+r+r,┌__call__LszJointDistribution.__call__)r+rlN)rrrsrtruZ _argnamesr'rvr-rFr┤ror╕r+r+r+r,rлs rлc@seZdZdZddДZdS)r7zg Representation of random symbols with joint probability distributions to allow indexing." cCsDt|jtГr@|jj|kdkr6td|j|jjdfГВt||ГSdS)NTz$Index keys for %s can only up to %s.r5)r&rGr$r=rP┌namer)r0┌keyr+r+r,┌__getitem__Ts zJointRandomSymbol.__getitem__N)rrrsrtrur╗r+r+r+r,r7Osr7c@sXeZdZdZddДZddДZeddДГZedd ДГZd dДZ dd ДZ ddДZddДZdS)┌MarginalDistributionzю Represents the marginal distribution of a joint probability space. Initialised using a probability distribution and random variables(or their indexed components) which should be a part of the resultant distribution. cGsВt|Гdkr$t|dГr$t|dГ}tddД|DГГsBttdГГВtаddД|DГб}t|t Гsttt |ГГdkrt|Stа|||бS)Nr5rcss|]}t|ttfГVqdSr%)r&rrrHr+r+r,rOirCz/MarginalDistribution.__new__..zБMarginal distribution can be intitialised only in terms of random variables or indexed random variablescss|] }|VqdSr%r+rHr+r+r,rOmrC) r:r!rS┌allrPr"rZfromiterr&rлrrr')r(r*rKr+r+r,r'fszMarginalDistribution.__new__cCsdSr%r+r/r+r+r,┌checkrszMarginalDistribution.checkcCs&ddД|jdDГ}tddД|DГОS)NcSsg|]}t|tГr|СqSr+)r&rr?r+r+r,rBwrCz,MarginalDistribution.set..r5cSsg|]}|jjСqSr+)rGr.rHr+r+r,rBxrC)r2r rJr+r+r,r.uszMarginalDistribution.setcCs|jd}ddД|DГS)Nr5cSsh|]}|jjТqSr+)rGr3rHr+r+r,┌ }rCz/MarginalDistribution.symbols..r1rJr+r+r,rоzs zMarginalDistribution.symbolscsШ|jd|jd}ЙЗfddДt|ГDГ}t|tГrpt|jjГ}tdddНЙtЗfdd Д|DГГ}|а|б}ntd d ДИDГГ}t ||а ||бГИОS)Nrr5csg|]}|Иvr|СqSr+r+r?r^r+r,rBБrCz,MarginalDistribution.pdf..┌xT)┌realc3s|]}tИ|ГVqdSr%r r?)r└r+r,rOЕrCz+MarginalDistribution.pdf..css*|]"}t|tГr|jjn|jdVqdSr░)r&rrGr3r2rHr+r+r,rOИrC)r2rr&rлr:r-rrSrFr┌compute_pdf)r0r└rb┌marginalise_outrZrer+)rKr└r,rFs zMarginalDistribution.pdfcCs4|D]*}d}t|tГr|jj}|а|||б}q|Sr4)r&rrGrFr├)r0rbrKrIZlpdfr+r+r,r┬Лs z MarginalDistribution.compute_pdfcCs░ddlm}t|tГr |jj}n$t|tГrD|jа|jа|j dбб}|а ||jjiб}|jjrrt ||jj|fГ}n:|jjrм|tjtjtjfvrЪ|j|jf}|||jj|fГ}|S)Nr)r r5)┌sympy.concrete.summationsr r&rrGr.rr_rMr2rXr3rVrrWrZIntegersZNaturalsZ Naturals0r▓┌sup)r0rbrIr ┌domr+r+r,r├Уs z$MarginalDistribution.marginalise_outcGs |j|ОSr%rмr╖r+r+r,r╕жszMarginalDistribution.__call__N) rrrsrtrur'r╛rvr.rоrFr┬r├r╕r+r+r+r,r╝]s r╝N)=ruZsympy.core.basicrZsympy.core.functionrZsympy.core.mulrZsympy.core.singletonrZsympy.core.symbolrrZsympy.core.sympifyrZsympy.sets.setsr Zsympy.tensor.indexedrZsympy.concrete.productsrr─r rZsympy.core.containersrZsympy.integrals.integralsrrZsympy.matricesrrrZsympy.stats.crvrrZsympy.stats.drvrrZsympy.stats.rvrrrrrrrr Zsympy.utilities.iterablesr!Zsympy.utilities.miscr"Zsympy.externalr#r$rwrЮrбr╡rлr7r╝r+r+r+r,┌s<(q''*¤<