&VfMdZddlZddlZddlmZdZejdGddZefd Z efd Z d Z dS) z"Package for histogram compression.N)Tuple) rii3i iii it$i'T)frozencDeZdZUdZeed<eed<deeeffdZdS)CompressedHistogramValuezRepresents a value in a compressed histogram. Attributes: basis_point: Compression point represented in basis point, 1/100th of a percent. value: Cumulative weight at the basis point. basis_pointvaluereturnc|j|jfS)z1Returns the basis point and the value as a tuple.)rr)selfs h/var/www/html/software/conda/lib/python3.11/site-packages/tensorboard/plugins/distribution/compressor.pyas_tuplez!CompressedHistogramValue.as_tuple*s $*--N)__name__ __module__ __qualname____doc__float__annotations__rr rr rrs[ LLL.%u -......rrc |js d|DStj|j}t |j}||dz|pdz }g}d}|t|kr tj |||d}|t|kr||}|dkr ||dz nd} || kr|dz }:|r| s|j } nt||dz |j } t|||j } t||| || | } | t||| |dz }nn|t|k |t|krX| t|||j |dz }|t|kX|S) aCreates fixed size histogram by adding compression to accumulated state. This routine transforms a histogram at a particular step by interpolating its variable number of buckets to represent their cumulative weight at a constant number of compression points. This significantly reduces the size of the histogram and makes it suitable for a two-dimensional area plot where the output of this routine constitutes the ranges for a single x coordinate. Args: histo: A HistogramProto object. bps: Compression points represented in basis points, 1/100ths of a percent. Defaults to normal distribution. Returns: List of values for each basis point. cRg|]$}t|d%Srr .0bs r z,compress_histogram_proto..C/III(C0099;;IIIr?rrightsider)numnparraybucketlist bucket_limitsumcumsumlen searchsortedminmax_lerpappendrr ) histobpsr)r+weightsvaluesjir- cumsum_prevlhsrhsweights r compress_histogram_protor>0s$ 9JIISIIII Xel # #F*++LB6::<<#637??AAG F A c#hh,, OGSV' : : :#g,,QZF,-EE'!a%..sK$$Q :K :i,q1u-uy99l1ouy11C3q6;SAAF MM23q66BBKKMM N N N FA  % c#hh,,& c#hh,, .s1vuyAAJJLLMMM Q c#hh,, Mrc.tj|}|js d|DS|dd|dd}}|dddf}t|dddf}||dz|pdz }g}d}|t |krtj|||d } | t |kr|| } | dkr || dz nd } | | kr| dz } :| r| s|} nt|| dz |} t|| |} t||| | | | }| t||| |dz }nn|t |k|t |krS| t||| |dz }|t |kS|S) aCreates fixed size histogram by adding compression to accumulated state. This routine transforms a histogram at a particular step by linearly interpolating its variable number of buckets to represent their cumulative weight at a constant number of compression points. This significantly reduces the size of the histogram and makes it suitable for a two-dimensional area plot where the output of this routine constitutes the ranges for a single x coordinate. Args: buckets: A list of buckets, each of which is a 3-tuple of the form `(min, max, count)`. bps: Compression points represented in basis points, 1/100ths of a percent. Defaults to normal distribution. Returns: List of values for each basis point. cRg|]$}t|d%Srrrs r rz&compress_histogram..xrrrr r%Nr!r"r#r)r'r(sizer*r,r-r.r/r1r0r2r3rr )bucketsr5minminmaxmaxcounts right_edgesr6resultbp_indexr9r-r:r;r<r=s r compress_histogramrJbs<(hwG <JIISIIII 1 wr{1~VV QQQT]Fwqqq!t}%%KB6::<<#637??AAG FH SXX   OGS] A A A#g,,QZF,-EE'!a%..sK$$Q 6K 6+a!e,f55k!nf--C3x=+vsCHHF MM(X??HHJJ    MH  ) SXX  * SXX   $S]F ; ; D D F F    A  SXX   MrcD|||z t||z z||z z zS)z)Affinely map from [x0, x1] onto [y0, y1].)r)xx0x1y0y1s r r2r2s) R5b>>)R"W5 55r) r dataclassesnumpyr'typingrNORMAL_HISTOGRAM_BPS dataclassrr>rJr2rrr rVs)( Kd###.......$#.$)=////d%98888v66666r