Zj8 SrSSKJr SSKrSSKJr SSKJrJrJrJ 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. )CallableN)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 o/root/GenerationalWealth/GenerationalWealth/venv/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 rrAffineTransformed.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__rrpropertyrr$r(__static_attributes__ __classcell__rs@rr r "sSt,tt ;; 77 $$rr c ^\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) rrr7r ModuleListvaluesLinearprojr8)rr6r7r8kwargsdimrs rrParameterProjection.__init__@sV "6" MM(//J["\J[3299[#>J["\] $#]s A-xclURVs/sH o""U5PM nnUR"U6$s snfN)r?r8)rrCr?params_unboundeds rforwardParameterProjection.forwardHs404 : DG : 011;s1)r7r8r?)r+r,r-r.intdictstrrtupletorchTensorrrGr0r1r2s@rr4r4?sl%%*.sCx.%FNsTYZ_ZfZfTgOgFh% %22% *=22rr4c.^\rSrSrU4SjrSrSrU=r$) LambdaLayerNc.>[TU]5 XlgrE)rrfunction)rrSrs rrLambdaLayer.__init__Os  rc(UR"U/UQ76$rErS)rrCargss rrGLambdaLayer.forwardSs}}Q&&&rrV)r+r,r-r.rrGr0r1r2s@rrPrPNs!''rrPc\rSrSr%\\S'\\S'\\\4\S'SS\SS4Sjjr S r SS \ RS-S \ RS-S\ 4S jjr\S\4S j5r\S\4Sj5r\S\4Sj5rS\S\R,4SjrS\ R4Sjr\S\ RS\ R4Sj5rSrg)DistributionOutputWdistribution_classr6r7rAr9Nc|XlURVs0sHo"XRU-_M snUlgs snfrE)rAr7)rrAks rrDistributionOutput.__init__\s3<@MMJMqC--"222MJ Js9crURS:XaUR"U6$[UR"U6S5$)NrAr\r)r distr_argss r_base_distribution%DistributionOutput._base_distribution`s6 88q=**J7 7t66 CQG Grrrc\URU5nUcUcU$[XBX0RS9$)Nr)rdr r)rrcrrdistrs 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. rar;)rAr 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. )lenrkr 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. )r6r7r8)r4r7rPr8)rr6s rget_parameter_projection+DistributionOutput.get_parameter_projections'##]]"4??3  rrWc[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)rrWs rr8DistributionOutput.domain_maps "##rrCchU[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@@)rMr'square)rCs r squareplusDistributionOutput.squarepluss* EJJu||A455<>"% %zz"~s{{2 b0AAArr;N)r+r,r-r.__doc__r7rJrKrIrr r\r classmethodrMrNr8r0r;rrrrse'(A>Hd38n>''BELLBu||BELLBBrrc\rSrSr%SrSSS.r\\\4\ S'\ r \ \ S'\ S\RS\R4S j5rS rg ) NormalOutputz# Normal distribution output class. rar*r7r\rrcURU5R[R"UR5R 5nUR S5UR S54$Nrr)rrrs rr8NormalOutput.domain_mapsJu%// EKK0H0L0LM{{2 b 111rr;N)r+r,r-r.rr7rJrKrIrr r\rrrMrNr8r0r;rrrrsS()15Hd38n5%%2U\\2%,,22rrc\rSrSr%SrSSS.r\\\4\ S'\ r \ \ S'\ S\RS\R4S j5rS \4S jrSS \RS -S\RS -S \4SjjrSrg )NegativeBinomialOutputz. Negative Binomial distribution output class. ra total_countlogitsr7r\rrchURU5nURS5URS54$r)r|r)rrrs rr8!NegativeBinomialOutput.domain_maps/nn[1 ""2&r(:::rr9czUup#URS:XaURX#S9$[URX#S9S5$)Nrarrb)rrcrrs rrd)NegativeBinomialOutput._base_distributionsE( 88q=**{*R Rt66;6^`ab brNrrcXUupEUbXSR5- nURXE45$rE)logrd)rrcrrrrs rrh#NegativeBinomialOutput.distributions4)   iik !F&& '<==rr;r~)r+r,r-r.rr7rJrKrIrrr\rrrMrNr8rrdrhr0r;rrrrs01A>Hd38n>//;U\\;5<<;;c cY] >$||d2 >BG,,QUBU >  > >rr)rcollections.abcrrMrtorch.distributionsrrrrr r r r rr4rPrZrrrr;rrrs% $/$: 2")) 2'"))'J=J=Z B' B 2% 2 >/ >r