謜i8 dZddlmZddlZddlmZddlmZmZmZm Z m Z m Z m Z Gdde Z Gdd ejZGd d ejZGd d ZGddeZGddeZGddeZy)z: Time series distributional output classes and utilities. )CallableN)nn)AffineTransform Distribution IndependentNegativeBinomialNormalStudentTTransformedDistributioncVeZdZddeffd ZedZedZedZxZ S)AffineTransformedbase_distributionc|dn||_|dn||_t| |t |j|j|gy)Ng?locscale event_dim)rrsuper__init__r)selfrrrr __class__s Z/mnt/e/genesis-system/.venv/lib/python3.12/site-packages/transformers/time_series_utils.pyrzAffineTransformed.__init__#sE!MSu +33 *_QUQ[Q[gp-q,rscb|jj|jz|jzS)z7 Returns the mean of the distribution. ) base_distmeanrrrs rrzAffineTransformed.mean)s& ~~""TZZ/$((::rcN|jj|jdzzS)z; Returns the variance of the distribution. )rvariancerrs rr!zAffineTransformed.variance0s! ~~&&Q66rc6|jjS)zE Returns the standard deviation of the distribution. )r!sqrtrs rstddevzAffineTransformed.stddev7s }}!!##r)NNr) __name__ __module__ __qualname__rrpropertyrr!r$ __classcell__rs@rr r "sMt,t ;; 77 $$rr c eZdZdedeeefdedeejfddffd Z dejdeejfd Z xZ S) ParameterProjection in_featuresargs_dim domain_map.returnNc t|di|||_tj|j Dcgc]}tj ||c}|_||_ycc}w)N) rrr.r ModuleListvaluesLinearprojr/)rr-r.r/kwargsdimrs rrzParameterProjection.__init__@sV "6"  MM(//J["\3299[##>"\] $#]sA*xch|jDcgc] }|| }}|j|Scc}wN)r6r/)rr9r6params_unboundeds rforwardzParameterProjection.forwardHs504 :DG::t 011;s/) r%r&r'intdictstrrtupletorchTensorrr=r)r*s@rr,r,?sg%%*.sCx.%FNsTYZ_ZfZfTgOgFh% %22% *=2rr,c$eZdZfdZdZxZS) LambdaLayerc0t|||_yr;)rrfunction)rrGrs rrzLambdaLayer.__init__Os   rc(|j|g|Sr;)rG)rr9argss rr=zLambdaLayer.forwardSst}}Q&&&r)r%r&r'rr=r)r*s@rrErENs !'rrEcdeZdZUeed<eed<eeefed<ddeddfdZdZ dd e jdzd e jdzde fd Z edefd Zedefd ZedefdZdedej*fdZde jfdZede jde jfdZy)DistributionOutputdistribution_classr-r.r8r0Nc|||_|jDcic]}|||j|zc}|_ycc}wr;)r8r.)rr8ks rrzDistributionOutput.__init__\s5<@MMJqC$--"222J Js9cp|jdk(r|j|St|j|dS)Nr8rLr)r distr_argss r_base_distributionz%DistributionOutput._base_distribution`s; 88q=*4**J7 76t66 CQG Grrrcb|j|}|||St||||jS)Nr)rSr r)rrRrrdistrs r distributionzDistributionOutput.distributionfs7 '' 3 ;5=L$U5DNN[ [rc>|jdk(rdS|jfS)zo Shape of each individual event contemplated by the distributions that this object constructs. rPr2)r8rs r event_shapezDistributionOutput.event_shapers XX]r3 3rc,t|jS)z Number of event dimensions, i.e., length of the `event_shape` tuple, of the distributions that this object constructs. )lenrXrs rrzDistributionOutput.event_dimys 4##$$rcy)z A float that will have a valid numeric value when computing the log-loss of the corresponding distribution. By default 0.0. This value will be used when padding data series. rr2rs rvalue_in_supportz#DistributionOutput.value_in_supports rcXt||jt|jS)z~ Return the parameter projection layer that maps the input to the appropriate parameters of the distribution. )r-r.r/)r,r.rEr/)rr-s rget_parameter_projectionz+DistributionOutput.get_parameter_projections'##]]"4??3  rrIct)a Converts arguments to the right shape and domain. The domain depends on the type of distribution, while the correct shape is obtained by reshaping the trailing axis in such a way that the returned tensors define a distribution of the right event_shape. )NotImplementedError)rrIs rr/zDistributionOutput.domain_maps "##rr9cd|tjtj|dzzdz S)z Helper to map inputs to the positive orthant by applying the square-plus operation. Reference: https://twitter.com/jon_barron/status/1387167648669048833 g@@)rBr#square)r9s r squarepluszDistributionOutput.squarepluss* EJJu||A455<r?r@rrSrBrCrrVr(rArXrfloatr\rModuler^r/ staticmethodrdr2rrrKrKWs38nKCKKH$(%) \\\D  \||d" \  \4U44 %3%%% C BII $ $=ell=u||==rrKceZdZUdZddddZeeefed<e Z e ed<e de jde jde jfd Zy ) StudentTOutputz. Student-T distribution output class. rP)dfrrr.rLrmrrc|j|jtj|jj }d|j|z}|j d|j d|j dfS)Nrbrd clamp_minrBfinfodtypeepssqueeze)clsrmrrs rr/zStudentTOutput.domain_mapsgu%// EKK0H0L0LM 3>>"% %zz"~s{{2 b0AAArN)r%r&r'__doc__r.r?r@r>rgr rLrf classmethodrBrCr/r2rrrlrlse'(A>Hd38n>''BELLBu||BELLBBrrlceZdZUdZdddZeeefed<e Z e ed<e de jde jfdZy ) NormalOutputz+ Normal distribution output class. rP)rrr.rLrrc|j|jtj|jj }|j d|j dfSNrorp)rvrrs rr/zNormalOutput.domain_mapsJu%// EKK0H0L0LM{{2 b 111rN)r%r&r'rwr.r?r@r>rgr rLrfrxrBrCr/r2rrrzrzsS()15Hd38n5%%2U\\2%,,22rrzceZdZUdZdddZeeefed<e Z e ed<e de jde jfdZd efd Z dd e jd zd e jd zd efdZy )NegativeBinomialOutputz6 Negative Binomial distribution output class. rP total_countlogitsr.rLrrch|j|}|jd|jdfSr|)rdru)rvrrs rr/z!NegativeBinomialOutput.domain_maps/nn[1 ""2&r(:::rr0c|\}}|jdk(r|j||St|j||dS)NrPrrQ)rrRrrs rrSz)NegativeBinomialOutput._base_distributionsL( V 88q=**{6*R Rt66;W]6^`ab brNrrc\|\}}|||jz }|j||fSr;)logrS)rrRrrrrs rrVz#NegativeBinomialOutput.distributions:) V   eiik !F&& V'<==rre)r%r&r'rwr.r?r@r>rgrrLrfrxrBrCr/rrSrVr2rrr~r~s01A>Hd38n>//;U\\;5<<;;c cY] >$||d2 >BG,,QUBU >  >rr~)rwcollections.abcrrBrtorch.distributionsrrrrr r r r rir,rErKrlrzr~r2rrrs% $/$: 2")) 2'"))'J=J=Z B' B 2% 2 >/ >r