...
", "loss2": "The lossy details
", } r" r )r r% r r. r markdown_to_safe_html)r r r resultr r( r s r _get_tag_to_descriptionr2 h s 0I0,, F" E 1# 67 | !&q/K5 1K "77 Ds E Mr c F | D ci c] }|t | | c}S c c}w )aM Returns a map of run names to a list of tag names. Args: mapping: a nested map `d` such that `d[run][tag]` is a time series produced by DataProvider's `list_*` methods. Returns: A map from run strings to a list of tag strings. E.g. {"loss001a": ["actor/loss", "critic/loss"], ...} )r% )r r s r _get_run_tag_infor4 s& 29 9#C % % 99 9s c 0 t | t | dS )a Prepares a scalar or histogram mapping for client consumption. Args: mapping: a nested map `d` such that `d[run][tag]` is a time series produced by DataProvider's `list_*` methods. Returns: A dict with the following fields: runTagInfo: the return type of `_get_run_tag_info` tagDescriptions: the return type of `_get_tag_to_description` ) runTagInfotagDescriptions)r4 r2 r s r _format_basic_mappingr9 s (027; r c |dz }g }| D ][ }t |j |k r |j |j |j |j | j d ] |S )a Formats image metadata from a list of BlobSequenceDatum's for clients. This expects that frontend clients need to access images based on the run+tag+sample. Args: sorted_datum_list: a list of DataProvider's `BlobSequenceDatum`, sorted by step. This can be produced via DataProvider's `read_blob_sequences`. sample: zero-indexed integer for the requested sample. Returns: A list of `ImageStepDatum` (see http_api.md). )stepwallTimeimageId)r valuesr' r< wall_timeblob_key)sorted_datum_listsampleindex step_datadatums r !_format_image_blob_sequence_datumrG sp QJEI" u|| % !OO <<.77 r c t j t }| j D ]4 \ }}|j D ] \ }}d|j dz i|| |< 6 t | S )a Returns a map of tag names to run information. Args: mapping: the result of DataProvider's `list_blob_sequences`. Returns: A nested map from run strings to tag string to image info, where image info is an object of form {"maxSamplesPerStep": num}. For example, { "reshaped": { "test": {"maxSamplesPerStep": 1}, "train": {"maxSamplesPerStep": 1} }, "convolved": {"test": {"maxSamplesPerStep": 50}}, } maxSamplesPerStepr; )r r dictr max_length)r tag_run_image_infor r r r s r _get_tag_run_image_inforM s{ " %006 &}} ^ , 2 2 4 NC#Y%9%9A%=, s #C ( " ##r c 0 t | t | dS )aV Prepares an image mapping for client consumption. Args: mapping: the result of DataProvider's `list_blob_sequences`. Returns: A dict with the following fields: tagRunSampledInfo: the return type of `_get_tag_run_image_info` tagDescriptions: the return type of `_get_tag_description_info` )r7 tagRunSampledInfo)r2 rM r8 s r _format_image_mappingrP s 37; 4W = r c 2 e Zd ZdZej Zd Zd Zd Z d Z d Z e j j d Zdd Zd Ze j j d Zd Zd Zd Zd Zd Zd Zd Zd Ze j j d Zd Z y) MetricsPluginz Metrics Plugin for TensorBoard.c |j | _ |j xs i }|j t j d |j t j d |j t j d d| _ t j dd | _ t j dd | _ t j d d | _ y ) zInstantiates MetricsPlugin. Args: context: A base_plugin.TBContext instance. MetricsLoader checks that it contains a valid `data_provider`. i 3 )scalars histogramsimageszscalar time seriesr ) data_kindlatest_known_versionzhistogram time serieszimage time seriesN) data_provider_data_providersampling_hintsgetscalar_metadata PLUGIN_NAMEhistogram_metadataimage_metadata_plugin_downsamplingr _MetadataVersionChecker_scalar_version_checker_histogram_version_checker_image_version_checker)selfcontextr] s r __init__zMetricsPlugin.__init__ s & 3 3 !//52%))/*E*EtL(,,".. %(( )C)CRH % ! (3'J'J*!"( $ +6*M*M-!"+ ' '2&I&I)!"' #r c 0 t j dd S )NTz Time Series)is_ng_componenttab_name)r FrontendMetadatarh s r frontend_metadataz MetricsPlugin.frontend_metadata s ++ = r c J | j | j | j dS )N)z/tagsz /timeSeriesz /imageData) _serve_tags_serve_time_series_serve_image_dataro s r get_plugin_appsz MetricsPlugin.get_plugin_apps s' %% 22 00 r c ` t j t j t j fS N)r_ r` ra rb ro s r data_plugin_namesz MetricsPlugin.data_plugin_names& s' ' ' * * & & r c y)NF ro s r is_activezMetricsPlugin.is_active- s r c t j |j }t j |j }| j || }t j ||d S )N) experimentapplication/json)r ri environ experiment_id _tags_implr Respond)rh requestctxr} rD s r rr zMetricsPlugin._serve_tags0 sR !!'//2 ..w? ; %1CDDr Nc x | j j ||t j }| j |t j | j }| j j ||t j }|i }| j |t j | j }| j j ||t j }|i }| j |t j | j }i }t | |d< t | |d< t | |d< |S )a Returns tag metadata for a given experiment's logged metrics. Args: ctx: A `tensorboard.context.RequestContext` value. experiment: optional string ID of the request's experiment. Returns: A nested dict 'd' with keys in ("scalars", "histograms", "images") and values being the return type of _format_*mapping. )r plugin_namerV rW rX )r\ list_scalarsr_ r` _filter_by_versionparse_plugin_metadatare list_tensorsra rf list_blob_sequencesrb rg r9 rP )rh r r} scalar_mappinghistogram_mapping image_mappingr1 s r r zMetricsPlugin._tags_impl7 sO ,,99 $'33 : 00 1 1 ( ( ! / / < < $*66 = $ " 3 3 4 4 + + ++?? $&22 @ M // 0 0 ' ' 1. Ay 45F G| 0 ?x r c |D ci c] }|i }}|j D ]U \ }}|j D ]= \ }} ||j } |j | j || s6||| |< ? W |S c c}w )z@Filter `DataProvider.list_*` output by summary metadata version.)r plugin_contentokversion) rh r parse_metadataversion_checkerr r1 r r r mds r r z MetricsPlugin._filter_by_versionm s %,-c#r'--#*==? - C "0"6"6"8 -Y#I$<$<=&))"**c3? #,s C - - .s A;c t j |j }t j |j }|j dk( r |j j d }n|j j d }|st j d t j | }| j ||| }t j ||d S # t $ r t j d w xY w)NPOSTrequestszMissing 'requests' fieldz"Unable to parse 'requests' as JSONr~ )r ri r r methodformr^ argsr InvalidArgumentErrorjsonloads ValueError_time_series_implr r )rh r r r} series_requests_stringseries_requestsresponses r rs z MetricsPlugin._serve_time_seriesx s !!'//2 ..w? >>V #%,\\%5%5j%A "%,\\%5%5j%A "%--.HI I "jj)? @O ))#z?K (4FGG --4 s C C;c P |D cg c] }| j ||| }}|S c c}w )a~ Constructs a list of responses from a list of series requests. Args: ctx: A `tensorboard.context.RequestContext` value. experiment: string ID of the request's experiment. series_requests: a list of `TimeSeriesRequest` dicts (see http_api.md). Returns: A list of `TimeSeriesResponse` dicts (see http_api.md). )_get_time_series)rh r r} r r responsess r r z MetricsPlugin._time_series_impl s= + ! !#z7 ; s #c |j d }|j d }|j d }|j d }||d}t |t r||d< t |t r||d< |S )Nr r pluginrC )r r )r^ isinstancestrint)rh series_requestr r r rC r s r _create_base_responsez#MetricsPlugin._create_base_response sx ' ' ##H- ##H- $S1 c3 !HUO fc "!'HX r c |j d }|j d }|j d }|j d }t |t sy|t j k7 r'|t j k7 r|t j k7 ry|t v rt |t sy|t v rt |t syy ) Nr r r rC z Missing tagzInvalid pluginz Missing runzMissing sample) r^ r r r_ r` ra rb _SINGLE_RUN_PLUGINS_SAMPLED_PLUGINSr )rh r r r r rC s r _get_invalid_request_errorz(MetricsPlugin._get_invalid_request_error s ' ##H- ' ##H-#s# o11 1,888.444# ( (C1E % %j.E#r c |j d }|j d }|j d }|j d }| j | }| j | } | r| |d< |S |r|gnd} d} |t j k( r| j |||| } |t j k( r| j |||| } |t j k( r| j ||||| } | |d< |S )aY Returns time series data for a given tag, plugin. Args: ctx: A `tensorboard.context.RequestContext` value. experiment: string ID of the request's experiment. series_request: a `TimeSeriesRequest` (see http_api.md). Returns: A `TimeSeriesResponse` dict (see http_api.md). r r r rC errorN runToSeries) r^ r r r_ r` _get_run_to_scalar_seriesra _get_run_to_histogram_seriesrb _get_run_to_image_series) rh r r} r r r r rC r request_errorr# run_to_seriess r r z MetricsPlugin._get_time_series s ' ' ##H- ##H---n= 77G -HW Ou _00 0 : :Zd M '33 3 = =Zd M ^// / 9 9Zfd M #0 r c b | j j ||t j | j d t j ||g }i }|j D ]C \ }}||vr || D cg c]& } | j | j | j d( } } | ||< E |S c c} w )a Builds a run-to-scalar-series dict for client consumption. Args: ctx: A `tensorboard.context.RequestContext` value. experiment: a string experiment id. tag: string of the requested tag. runs: optional list of run names as strings. Returns: A map from string run names to `ScalarStepDatum` (see http_api.md). rV r# tagsr r downsamplerun_tag_filter)r= r< value) r\ read_scalarsr_ r` rc r RunTagFilterr r@ r< r rh r r} r r# r r result_runtag_datarF r? s r r z'MetricsPlugin._get_run_to_scalar_series s %%22 $'3300;#00d#G 3 $+MMO / J(" &c] !& !JJ "[[F )/M* % / s 6+B,c | |j j }|D cg c] }|d |d |d d }}|S c c}w )zFormats a histogram datum's bins for client consumption. Args: datum: a DataProvider's TensorDatum. Returns: A list of `HistogramBin`s (see http_api.md). r r" r; )minmaxcount)numpytolist)rh rF numpy_listxbinss r _format_histogram_datum_binsz*MetricsPlugin._format_histogram_datum_bins sG [['') CMNa!QqTAaD9NN Os 9c l | j j ||t j | j d t j ||g }i }|j D ]H \ }}||vr || D cg c]+ } | j | j | j | d- } } | ||< J |S c c} w )a Builds a run-to-histogram-series dict for client consumption. Args: ctx: A `tensorboard.context.RequestContext` value. experiment: a string experiment id. tag: string of the requested tag. runs: optional list of run names as strings. Returns: A map from string run names to `HistogramStepDatum` (see http_api.md). rW r r )r= r< r ) r\ read_tensorsra r` rc r r r r@ r< r r s r r z*MetricsPlugin._get_run_to_histogram_series s %%22 $*6600 >#00d#G 3 $+MMO / J(" &c] !& !JJ = =e DF )/M* % / s 60B1c | j j ||t j | j d t j ||g }i }|j D ]# \ }} || vr | | } t | | } | s | ||< % |S )a Builds a run-to-image-series dict for client consumption. Args: ctx: A `tensorboard.context.RequestContext` value. experiment: a string experiment id. tag: string of the requested tag. sample: zero-indexed integer for the requested sample. runs: optional list of run names as strings. Returns: A `RunToSeries` dict (see http_api.md). rX )r r ) r\ read_blob_sequencesrb r` rc r r r rG ) rh r r} r rC r# r r r r blob_sequence_datum_listseriess r r z&MetricsPlugin._get_run_to_image_series@ s %%99 $&2200:#00SEB : $+MMO 3 J("'/} $6(&F ,2 j) 3 r c t j |j }|j d }|st j d | j || \ }}t j ||| S )zServes an individual image.r> zMissing 'imageId' field) r ri r r r r _image_data_implr r )rh r r rA data content_types r rt z MetricsPlugin._serve_image_datab sg !!'//2<< *--.GH H # 4 4S( C | $ ==r c | j j || }t j d| }t j |t }||fS )ao Gets the image data for a blob key. Args: ctx: A `tensorboard.context.RequestContext` value. blob_key: a string identifier for a DataProvider blob. Returns: A tuple containing: data: a raw bytestring of the requested image's contents. content_type: a string HTTP content type. )rA N)r\ read_blobimghdrwhat_IMGHDR_TO_MIMETYPEr^ _DEFAULT_IMAGE_MIMETYPE)rh r rA r image_typer s r r z MetricsPlugin._image_data_implm sR "",,S8,D[[t, *.. / l##r rw ) __name__ __module__ __qualname____doc__r r` r rj rp ru rx r{ r Request applicationrr r r rs r r r r r r r r rt r rz r r rR rR s )&&K B !!E "E 4l !!H "H&" 0'R" H " H D !!> ">$r rR )'r r r r werkzeugr tensorboardr r tensorboard.backendr tensorboard.datar tensorboard.pluginsr tensorboard.plugins.histogramr ra tensorboard.plugins.imagerb tensorboard.plugins.metricstensorboard.plugins.scalarr_ r r frozensetr` r r r r. r2 r4 r9 rG rM rP TBPluginrR rz r r