o  hE@sdZddlmZmZddlZddlmZddlmZmZm Z m Z m Z m Z m Z Gddde ZGdd d ejZGd d d ejZGd d d ZGdddeZGdddeZGdddeZdS)z: Time series distributional output classes and utilities. )CallableOptionalN)nn)AffineTransform Distribution IndependentNegativeBinomialNormalStudentTTransformedDistributioncsHeZdZd deffdd ZeddZedd Zed d ZZ S) AffineTransformedNrbase_distributioncsF|durdn||_|durdn||_t|t|j|j|dgdS)Ng?locscale event_dim)rrsuper__init__r)selfr rrr __class__r/var/www/html/construction_image-detection-poc/venv/lib/python3.10/site-packages/transformers/time_series_utils.pyr#s"zAffineTransformed.__init__cCs|jj|j|jS)z7 Returns the mean of the distribution. ) base_distmeanrrrrrrr)szAffineTransformed.meancCs|jj|jdS)z; Returns the variance of the distribution. )rvariancerrrrrr0szAffineTransformed.variancecCs |jS)zE Returns the standard deviation of the distribution. )rsqrtrrrrstddev7s zAffineTransformed.stddev)NNr) __name__ __module__ __qualname__rrpropertyrrr __classcell__rrrrr "s  r c s^eZdZdedeeefdedeej fddffdd Z d ej deej fd d Z Z S) ParameterProjection in_featuresargs_dim domain_map.returnNc s@tjdi|||_tfdd|D|_||_dS)Ncsg|]}t|qSr)rLinear).0dimr'rr Esz0ParameterProjection.__init__..r)rrr(r ModuleListvaluesprojr))rr'r(r)kwargsrr.rr@s zParameterProjection.__init__xcsfdd|jD}|j|S)Ncsg|]}|qSrr)r,r2r4rrr/Isz/ParameterProjection.forward..)r2r))rr4params_unboundedrr5rforwardHs zParameterProjection.forward) r!r"r#intdictstrrtupletorchTensorrr7r%rrrrr&?s "r&cs$eZdZfddZddZZS) LambdaLayercst||_dSN)rrfunction)rr@rrrrOs  zLambdaLayer.__init__cGs|j|g|RSr?)r@)rr4argsrrrr7SszLambdaLayer.forward)r!r"r#rr7r%rrrrr>Ns r>c@seZdZUeed<eed<eeefed<ddeddfdd Zd d Z  dd e e j d e e j de fddZedefddZedefddZedefddZdedejfddZde j fddZede j de j fddZdS) DistributionOutputdistribution_classr'r(r-r*Ncs"_fddjD_dS)Ncsi|] }|j|qSr)r()r,kr-rrr ^sz/DistributionOutput.__init__..)r-r()rr-rrFrr\szDistributionOutput.__init__cCs$|jdkr |j|St|j|dS)NrDr-rCr)r distr_argsrrr_base_distribution`s  z%DistributionOutput._base_distributionrrcCs0||}|dur|dur|St||||jdS)Nr)rJr r)rrIrrdistrrrr distributionfs zDistributionOutput.distributioncCs|jdkrdS|jfS)zo Shape of each individual event contemplated by the distributions that this object constructs. rDr)r-rrrr event_shaperszDistributionOutput.event_shapecCs t|jS)z Number of event dimensions, i.e., length of the `event_shape` tuple, of the distributions that this object constructs. )lenrMrrrrrys zDistributionOutput.event_dimcCsdS)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. rrrrrrvalue_in_supportsz#DistributionOutput.value_in_supportcCst||jt|jdS)z~ Return the parameter projection layer that maps the input to the appropriate parameters of the distribution. )r'r(r))r&r(r>r))rr'rrrget_parameter_projections z+DistributionOutput.get_parameter_projectionrAcGst)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)rrArrrr)szDistributionOutput.domain_mapr4cCs|tt|ddS)z Helper to map inputs to the positive orthant by applying the square-plus operation. Reference: https://twitter.com/jon_barron/status/1387167648669048833 g@@)r<rsquarer5rrr squareplusszDistributionOutput.squareplus)rDNN)r!r"r#type__annotations__r8r9r:rrJrr<r=rrLr$r;rMrfloatrOrModulerPr) staticmethodrTrrrrrBWs2     rBc@sZeZdZUdZddddZeeefed<e Z e ed<e de jde jde jfd d Zd S) StudentTOutputz. Student-T distribution output class. rD)dfrrr(rCr\rrcCsD||t|jj}d||}|d|d|dfS)NrRrT clamp_minr<finfodtypeepssqueeze)clsr\rrrrrr)szStudentTOutput.domain_mapN)r!r"r#__doc__r(r9r:r8rWr rCrV classmethodr<r=r)rrrrr[s  "r[c@sReZdZUdZdddZeeefed<e Z e ed<e de jde jfdd Zd S) NormalOutputz+ Normal distribution output class. rD)rrr(rCrrcCs.||t|jj}|d|dfSNr]r^)rdrrrrrr)szNormalOutput.domain_mapN)r!r"r#rer(r9r:r8rWr rCrVrfr<r=r)rrrrrgs  rgc@seZdZUdZdddZeeefed<e Z e ed<e de jde jfdd Zd efd d Z ddee jdee jd efddZd S)NegativeBinomialOutputz6 Negative Binomial distribution output class. rD total_countlogitsr(rCrkrlcCs||}|d|dfSrh)rTrc)rdrkrlrrrr)s z!NegativeBinomialOutput.domain_mapr*cCs4|\}}|jdkr|j||dSt|j||ddS)NrDrjrH)rrIrkrlrrrrJs z)NegativeBinomialOutput._base_distributionNrrcCs*|\}}|dur||7}|||fSr?)logrJ)rrIrrrkrlrrrrLs z#NegativeBinomialOutput.distributionrU)r!r"r#rer(r9r:r8rWrrCrVrfr<r=r)rrJrrLrrrrris   ri)retypingrrr<rtorch.distributionsrrrrr r r r rYr&r>rBr[rgrirrrrs $  M