ga0dZddlmZmZmZmZddlZddlmZddlm Z m Z m Z m Z m Z mZmZGddeZGdd ejZGd d ejZGd d ZGddeZGddeZGddeZdS)z: Time series distributional output classes and utilities. )CallableDictOptionalTupleN)nn)AffineTransform Distribution IndependentNegativeBinomialNormalStudentTTransformedDistributioncheZdZddeffd ZedZedZedZxZ S) AffineTransformedNrbase_distributionc|dn||_|dn||_t|t |j|j|gdS)Ng?locscale event_dim)rrsuper__init__r)selfrrrr __class__s Z/var/www/html/ai-engine/env/lib/python3.11/site-packages/transformers/time_series_utils.pyrzAffineTransformed.__init__$s[!MSSu +333 *_QUQ[gp-q-q-q,rsssssc:|jj|jz|jzS)z7 Returns the mean of the distribution. ) base_distmeanrrrs rr zAffineTransformed.mean*s ~"TZ/$(::rc0|jj|jdzzS)z; Returns the variance of the distribution. )rvariancerr!s rr$zAffineTransformed.variance1s ~&Q66rc4|jS)zE Returns the standard deviation of the distribution. )r$sqrtr!s rstddevzAffineTransformed.stddev8s }!!###r)NNr) __name__ __module__ __qualname__r rpropertyr r$r' __classcell__rs@rrr#stt,tttttt ;;X; 77X7 $$X$$$$$rrc eZdZdedeeefdedeej fddffd Z dej deej fd Z xZ S) ParameterProjection in_featuresargs_dim domain_map.returnNc tjdi|||_tjfd|D|_||_dS)Nc:g|]}tj|S)rLinear).0dimr0s r z0ParameterProjection.__init__..Fs%"\"\"\329[##>#>"\"\"\rr6)rrr1r ModuleListvaluesprojr2)rr0r1r2kwargsrs ` rrzParameterProjection.__init__Ase ""6"""  M"\"\"\"\(//J[J["\"\"\]] $rxc>fd|jD}|j|S)Nc&g|] }|Sr6r6)r8r=r?s rr:z/ParameterProjection.forward..Js!:::DDGG:::r)r=r2)rr?params_unboundeds ` rforwardzParameterProjection.forwardIs/:::: :::t 011r) r(r)r*intrstrrrtorchTensorrrCr,r-s@rr/r/@s%%*.sCx.%FNsTYZ_ZfTgOgFh% %%%%%%22% *=22222222rr/c$eZdZfdZdZxZS) LambdaLayercVt||_dSN)rrfunction)rrLrs rrzLambdaLayer.__init__Ps$   rc|j|g|RSrK)rL)rr?argss rrCzLambdaLayer.forwardTst}Q&&&&&r)r(r)r*rrCr,r-s@rrIrIOsG!!!!!'''''''rrIcTeZdZUeed<eed<eeefed<ddeddfdZd Z dd e e j d e e j de fd Zedefd ZedefdZedefdZdedejfdZde j fdZede j de j fdZdS)DistributionOutputdistribution_classr0r1r9r3NcJ_fdjD_dS)Nc2i|]}|j|zSr6)r1)r8kr9rs r z/DistributionOutput.__init__.._s'JJJqC$-"22JJJr)r9r1)rr9s``rrzDistributionOutput.__init__]s/JJJJJDMJJJ rc\|jdkr |j|St|j|dS)NrRr9rQr )r distr_argss r_base_distributionz%DistributionOutput._base_distributionas8 8q==*4*J7 76t6 CQGG Grrrch||}|||St||||jS)Nr)rZrr)rrYrrdistrs r distributionzDistributionOutput.distributiongs> '' 33 ;5=L$U5DN[[[ [rc,|jdkrdn|jfS)zo Shape of each individual event contemplated by the distributions that this object constructs. rRr6)r9r!s r event_shapezDistributionOutput.event_shapess X]]rr 3rc*t|jS)z Number of event dimensions, i.e., length of the `event_shape` tuple, of the distributions that this object constructs. )lenr_r!s rrzDistributionOutput.event_dimzs 4#$$$rcdS)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. rr6r!s rvalue_in_supportz#DistributionOutput.value_in_supports srcTt||jt|jS)z~ Return the parameter projection layer that maps the input to the appropriate parameters of the distribution. )r0r1r2)r/r1rIr2)rr0s rget_parameter_projectionz+DistributionOutput.get_parameter_projections0##]"4?33    rrNct)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)rrNs rr2zDistributionOutput.domain_maps "###rr?c`|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@@)rFr&square)r?s r squarepluszDistributionOutput.squarepluss+ EJu|A4555<>"%% %zz"~~s{{2 b0A0AAArN)r(r)r*__doc__r1rrErDrnr rQrm classmethodrFrGr2r6rrrsrss'(A>>Hd38n>>>''''BELBu|BELBBB[BBBrrsczeZdZUdZdddZeeefed<e Z e ed<e de jde jfdZd S) NormalOutputz+ Normal distribution output class. rR)rrr1rQrrc||tj|jj}|d|dfSNrvrw)r}rrs rr2zNormalOutput.domain_mapsQu%%// EK0H0H0LMM{{2 b 1 111rN)r(r)r*r~r1rrErDrnr rQrmrrFrGr2r6rrrrs()155Hd38n555%%%%2U\2%,222[222rrceZdZUdZdddZeeefed<e Z e ed<e de jde jfdZd efd Z dd ee jd ee jd efdZd S)NegativeBinomialOutputz6 Negative Binomial distribution output class. rR total_countlogitsr1rQrrc||}|d|dfSr)rkr|)r}rrs rr2z!NegativeBinomialOutput.domain_maps8nn[11 ""2&&r(:(:::rr3c|\}}|jdkr|||St|||dS)NrRrrX)rrYrrs rrZz)NegativeBinomialOutput._base_distributionsU( V 8q==**{6*RR Rt66;W]6^^`abb brNrrcl|\}}|||z }|||fSrK)logrZ)rrYrrrrs rr]z#NegativeBinomialOutput.distributions?) V   eiikk !F&& V'<===rrl)r(r)r*r~r1rrErDrnr rQrmrrFrGr2r rZrr]r6rrrrs01A>>Hd38n>>>////;U\;5<;;;[;c cccc_c > >' 5 >EMelE[ >  > > > > > >rr)r~typingrrrrrFrtorch.distributionsrr r r r r rrrpr/rIrPrsrrr6rrrs 322222222222 $$$$$/$$$: 2 2 2 2 2") 2 2 2'''''")'''J=J=J=J=J=J=J=J=Z B B B B B' B B B 2 2 2 2 2% 2 2 2 > > > > >/ > > > > >r