eThESrSSKJrJr SSKrSSKJr SSKJrJrJ r J r J r J r J r "SS\ 5r"SS \R5r"S S \R5r"S S 5r"SS\5r"SS\5r"SS\5rg)z: Time series distributional output classes and utilities. )CallableOptionalN)nn)AffineTransform Distribution IndependentNegativeBinomialNormalStudentTTransformedDistributioncd^\rSrSrSS\4U4Sjjjr\S5r\S5r\S5r Sr U=r $) AffineTransformed"base_distributionc >UcSOUUlUcSOUUl[TU] U[ URURUS9/5 g)Ng?locscale event_dim)rrsuper__init__r)selfrrrr __class__s V/var/www/auris/envauris/lib/python3.13/site-packages/transformers/time_series_utils.pyrAffineTransformed.__init__#sE!MSu +33 *_QUQ[Q[gp-q,rscbURRUR-UR-$)z' Returns the mean of the distribution. ) base_distmeanrrrs rr AffineTransformed.mean)s& ~~""TZZ/$((::rcNURRURS--$)z+ Returns the variance of the distribution. )rvariancerr!s rr%AffineTransformed.variance0s! ~~&&Q66rc6URR5$)z5 Returns the standard deviation of the distribution. )r%sqrtr!s rstddevAffineTransformed.stddev7s }}!!##rrr)NNr) __name__ __module__ __qualname____firstlineno__rrpropertyr r%r)__static_attributes__ __classcell__rs@rrr"sSt,tt ;; 77 $$rrc ^\rSrSrS\S\\\4S\S\\ R4SS4U4Sjjr S \ RS\\ R4S jr S r U=r$) ParameterProjection? in_featuresargs_dim domain_map.returnNc >[TU]"S0UD6 X l[R"UR 5Vs/sHn[R "X5PM sn5UlX0lgs snf)N) rrr8r ModuleListvaluesLinearprojr9)rr7r8r9kwargsdimrs rrParameterProjection.__init__@sV "6" MM(//J["\J[3299[#>J["\] $#]s A-xclURVs/sH o""U5PM nnUR"U6$s snfN)r@r9)rrDr@params_unboundeds rforwardParameterProjection.forwardHs404 : DG : 011;s1)r8r9r@)r,r-r.r/intdictstrrtupletorchTensorrrHr1r2r3s@rr5r5?sl%%*.sCx.%FNsTYZ_ZfZfTgOgFh% %22% *=22rr5c.^\rSrSrU4SjrSrSrU=r$) LambdaLayerNc.>[TU]5 XlgrF)rrfunction)rrTrs rrLambdaLayer.__init__Os  rc(UR"U/UQ76$rFrT)rrDargss rrHLambdaLayer.forwardSs}}Q&&&rrW)r,r-r.r/rrHr1r2r3s@rrQrQNs!''rrQc\rSrSr%\\S'\\S'\\\4\S'SS\SS4Sjjr S r SS \ \ RS \ \ RS\4S jjr\S\4S j5r\S\4Sj5r\S\4Sj5rS\S\R.4SjrS\ R4Sjr\S\ RS\ R4Sj5rSrg)DistributionOutputWdistribution_classr7r8rBr:Nc|XlURVs0sHo"XRU-_M snUlgs snfrF)rBr8)rrBks rrDistributionOutput.__init__\s3<@MMJMqC--"222MJ Js9crURS:XaUR"U6$[UR"U6S5$)NrBr]r)r distr_argss r_base_distribution%DistributionOutput._base_distribution`s6 88q=**J7 7t66 CQG Grrrc\URU5nUcUcU$[XBX0RS9$)Nr)rerr)rrdrrdistrs r distributionDistributionOutput.distributionfs3 '' 3 ;5=L$U5NN[ [rc@URS:XaS$UR4$)z_ Shape of each individual event contemplated by the distributions that this object constructs. rbr<)rBr!s r event_shapeDistributionOutput.event_shapers XX]r3 3rc,[UR5$)zx Number of event dimensions, i.e., length of the `event_shape` tuple, of the distributions that this object constructs. )lenrlr!s rrDistributionOutput.event_dimys 4##$$rcg)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. rr<r!s rvalue_in_support#DistributionOutput.value_in_supports rcR[UUR[UR5S9$)zn Return the parameter projection layer that maps the input to the appropriate parameters of the distribution. )r7r8r9)r5r8rQr9)rr7s rget_parameter_projection+DistributionOutput.get_parameter_projections'##]]"4??3  rrXc[5e)z 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)rrXs rr9DistributionOutput.domain_maps "##rrDchU[R"[R"U5S-5-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@@)rNr(square)rDs r squareplusDistributionOutput.squarepluss* EJJu||A455<>"% %zz"~s{{2 b0AAArr<N)r,r-r.r/__doc__r8rKrLrJrr r]r classmethodrNrOr9r1r<rrrrse'(A>Hd38n>''BELLBu||BELLBBrrc\rSrSr%SrSSS.r\\\4\ S'\ r \ \ S'\ S\RS\R4S j5rS rg ) NormalOutputz# Normal distribution output class. rbr+r8r]rrcURU5R[R"UR5R 5nUR S5UR S54$Nrr)rrrs rr9NormalOutput.domain_mapsJu%// EKK0H0L0LM{{2 b 111rr<N)r,r-r.r/rr8rKrLrJrr r]rrrNrOr9r1r<rrrrsS()15Hd38n5%%2U\\2%,,22rrc\rSrSr%SrSSS.r\\\4\ S'\ r \ \ S'\ S\RS\R4S j5rS \4S jrSS \\RS\\RS \4SjjrSrg )NegativeBinomialOutputz. Negative Binomial distribution output class. rb total_countlogitsr8r]rrchURU5nURS5URS54$r)r}r)rrrs rr9!NegativeBinomialOutput.domain_maps/nn[1 ""2&r(:::rr:czUup#URS:XaURX#S9$[URX#S9S5$)Nrbrrc)rrdrrs rre)NegativeBinomialOutput._base_distributionsE( 88q=**{*R Rt66;6^`ab brNrrcXUupEUbXSR5- nURXE45$rF)logre)rrdrrrrs rri#NegativeBinomialOutput.distributions4)   iik !F&& '<==rr<r)r,r-r.r/rr8rKrLrJrr r]rrrNrOr9rrerrir1r<rrrrs01A>Hd38n>//;U\\;5<<;;c c_c >' 5 >EMellE[ >  > >rr)rtypingrrrNrtorch.distributionsrrrr r r r rrr5rQr[rrrr<rrrs& $/$: 2")) 2'"))'J=J=Z B' B 2% 2 >/ >r