o 2hM@sbdZddlZddlZddlmZdZejddGdddZefd d Z efd d Z d dZ dS)z"Package for histogram compression.N)Tuple) rii3i iii it$i'T)frozenc@s8eZdZUdZeed<eed<deeeffddZdS)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_pointvaluereturncCs |j|jfS)z1Returns the basis point and the value as a tuple.)rr)selfr f/var/www/html/chatgem/venv/lib/python3.10/site-packages/tensorboard/plugins/distribution/compressor.pyas_tuple*s z!CompressedHistogramValue.as_tupleN)__name__ __module__ __qualname____doc__float__annotations__rr r r r r rs rc Csb|js dd|DSt|j}t|j}||d|pd}g}d}|t|krtj |||dd}|t|kr||}|dkrK||dnd } || krV|d7}q7|rZ| s^|j } n t ||d|j } t |||j } t ||| || | } | t||| |d7}nn|t|ks-|t|kr| t|||j |d7}|t|ks|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. cSg|] }t|dqSrr .0br r r Cz,compress_histogram_proto..?rrightsider)numnparraybucketlist bucket_limitsumcumsumlen searchsortedminmax_lerpappendrr ) histobpsr$r&weightsvaluesjir( cumsum_prevlhsrhsweightr r r compress_histogram_proto0s<      r9cCst|}|jsdd|DS|dd|dd}}|dddf}t|dddf}||d|p8d}g}d}|t|krtj|||d d } | t|kr|| } | dkrd|| dnd } | | kro| d7} qP| rs| sv|} n t|| d|} t || |} t ||| | | | }| t ||| |d7}nn|t|ksF|t|kr| t ||| |d7}|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. cSrrrrr r r rxrz&compress_histogram..rrr Nrrrr)r"r#sizer%r'r(r)r*r,r+r-r.rr )bucketsr0minminmaxmaxcounts right_edgesr1resultbp_indexr4r(r5r6r7r8r r r compress_histogrambsH     rCcCs |||t||||S)z)Affinely map from [x0, x1] onto [y0, y1].)r)xx0x1y0y1r r r r-s r-) r dataclassesnumpyr"typingrNORMAL_HISTOGRAM_BPS dataclassrr9rCr-r r r r s    2 ;