U ,-e`@sdZddlmZmZmZmZddlZddlmZddlm Z m Z m Z m Z m Z mZmZGdddeZGdd d ejZGd d d ejZGd d d ZGdddeZGdddeZGdddeZdS)z: Time series distributional output classes and utilities. )CallableDictOptionalTupleN)nn)AffineTransform Distribution IndependentNegativeBinomialNormalStudentTTransformedDistributioncsHeZdZd edfdd ZeddZedd Zed d ZZ S) AffineTransformedNr)base_distributioncsF|dkr dn||_|dkrdn||_t|t|j|j|dgdS)Ng?locscale event_dim)rrsuper__init__r)selfrrrr __class___/var/www/html/Darija-Ai-Train/env/lib/python3.8/site-packages/transformers/time_series_utils.pyr#szAffineTransformed.__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. )rvariancerrrrrr 0szAffineTransformed.variancecCs |jS)zE Returns the standard deviation of the distribution. )r sqrtrrrrstddev7szAffineTransformed.stddev)NNr) __name__ __module__ __qualname__rrpropertyrr r" __classcell__rrrrr"s  rcsVeZdZeeeefedeej fddfdd Z ej eej dddZ Z S) ParameterProjection.N) in_featuresargs_dim domain_mapreturnc s<tjf|||_tfdd|D|_||_dS)Ncsg|]}t|qSr)rZLinear).0dimr)rr Esz0ParameterProjection.__init__..)rrr*rZ ModuleListvaluesprojr+)rr)r*r+kwargsrr/rr@szParameterProjection.__init__xr,csfdd|jD}|j|S)Ncsg|] }|qSrr)r-r2r5rrr0Isz/ParameterProjection.forward..)r2r+)rr5Zparams_unboundedrr6rforwardHszParameterProjection.forward) r#r$r%intrstrrrtorchTensorrr7r'rrrrr(?s  r(cs$eZdZfddZddZZS) LambdaLayercst||_dSN)rrfunction)rr>rrrrOs zLambdaLayer.__init__cGs|j|f|Sr=)r>)rr5argsrrrr7SszLambdaLayer.forward)r#r$r%rr7r'rrrrr<Ns r<c@seZdZUeed<eed<eeefed<deddddZd d Z de e j e e j e d d d ZeedddZeedddZeedddZeejdddZe j dddZee j e j dddZdS) DistributionOutputdistribution_classr)r*N)r.r,cs"_fddjD_dS)Ncsi|]}|j|qSr)r*)r-kr.rrr ^sz/DistributionOutput.__init__..)r.r*)rr.rrDrr\szDistributionOutput.__init__cCs(|jdkr|j|St|j|dSdS)NrBr.rAr )r distr_argsrrr_base_distribution`s  z%DistributionOutput._base_distributionrrr,cCs4||}|dkr|dkr|St||||jdSdS)Nr)rHrr)rrGrrZdistrrrr distributionfs zDistributionOutput.distributionr,cCs|jdkrdS|jfS)zo Shape of each individual event contemplated by the distributions that this object constructs. rBr)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. )lenrLrrrrryszDistributionOutput.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_support)r)r,cCst||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_projection)r?cGs tdS)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. N)NotImplementedError)rr?rrrr+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:r!Zsquarer6rrr squareplusszDistributionOutput.squareplus)rB)NN)r#r$r%type__annotations__r8rr9rrHrr:r;rrJr&rrLrfloatrNrModulerOr+ staticmethodrRrrrrr@Ws,    r@c@sVeZdZUdZddddZeeefed<e Z e ed<e e je je jdddZdS) StudentTOutputz. Student-T distribution output class. rB)dfrrr*rAcCsD||t|jj}d||}|d|d|dfS)NrQrRZ clamp_minr:ZfinfoZdtypeZepssqueeze)clsrYrrrrrr+szStudentTOutput.domain_mapN)r#r$r%__doc__r*rr9r8rTr rArS classmethodr:r;r+rrrrrXs  rXc@sPeZdZUdZdddZeeefed<e Z e ed<e e je jdddZdS) NormalOutputz+ Normal distribution output class. rB)rrr*rAcCs.||t|jj}|d|dfSNrZr[)r]rrrrrr+szNormalOutput.domain_mapN)r#r$r%r^r*rr9r8rTr rArSr_r:r;r+rrrrr`s  r`c@s~eZdZUdZdddZeeefed<e Z e ed<e e je jdddZedd d Zdee jee jed d dZd S)NegativeBinomialOutputz6 Negative Binomial distribution output class. rB total_countlogitsr*rAcCs||}|d|dfSra)rRr\)r]rdrerrrr+s z!NegativeBinomialOutput.domain_maprKcCs8|\}}|jdkr |j||dSt|j||ddSdS)NrBrcrF)rrGrdrerrrrHs z)NegativeBinomialOutput._base_distributionNrIcCs*|\}}|dk r||7}|||fSr=)logrH)rrGrrrdrerrrrJs z#NegativeBinomialOutput.distribution)NN)r#r$r%r^r*rr9r8rTr rArSr_r:r;r+rrHrrJrrrrrbs   rb)r^typingrrrrr:rZtorch.distributionsrrr r r r r rrVr(r<r@rXr`rbrrrrs $  M