1Vh#ddlZddlmZddlmZddlmZddlm Z ddl m Z ddl m Z dZGd d e jZy) N)wrappers) plugin_util)provider) http_util) base_plugin)metadatadceZdZdZej ZdZejjdZ dZ ejjdZ dZdZdZd Zd Zd Zy ) PrCurvesPluginz6A plugin that serves PR curves for individual classes.c|j|_|jxsijtj t |_tjdd|_ y)zInstantiates a PrCurvesPlugin. Args: context: A base_plugin.TBContext instance. A magic container that TensorBoard uses to make objects available to the plugin. zPR curver) data_kindlatest_known_versionN) data_provider_data_providersampling_hintsgetr PLUGIN_NAME_DEFAULT_DOWNSAMPLING_downsample_tor_MetadataVersionChecker_version_checker)selfcontexts ]/home/dcms/DCMS/lib/python3.12/site-packages/tensorboard/plugins/pr_curve/pr_curves_plugin.py__init__zPrCurvesPlugin.__init__"sW&33&55;@@  "7 !, C C !"! c tj|j}tj|j}|jj d}|st j|ddS|jjd}|st j|ddS t j||j||||d}|S#t$r+}t j|t|ddcYd}~Sd}~wwxYw) aA route that returns a JSON mapping between runs and PR curve data. Returns: Given a tag and a comma-separated list of runs (both stored within GET parameters), fetches a JSON object that maps between run name and objects containing data required for PR curves for that run. Runs that either cannot be found or that lack tags will be excluded from the response. runz,No runs provided when fetching PR curve dataitagz+No tag provided when fetching PR curve dataapplication/jsonz text/plainN) rrenviron experiment_idargsgetlistrRespondrpr_curves_impl ValueErrorstr)rrequestctx experimentrunsrresponsees rpr_curves_routezPrCurvesPlugin.pr_curves_route2s!!'//2 ..w? ||##E*$$G llu%$$F  I ((##CT3?"H I$$Wc!flCH H Is()C D D<DDcli}tj||g}|jj||tj ||j }|D]X}|j|ij|} | td|d|| D cgc]} |j| c} ||<Z|Scc} w)a}Creates the JSON object for the PR curves response for a run-tag combo. Arguments: runs: A list of runs to fetch the curves for. tag: The tag to fetch the curves for. Raises: ValueError: If no PR curves could be fetched for a run and tag. Returns: The JSON object for the PR curves route response. )r" plugin_namerun_tag_filter downsamplez$No PR curves could be found for run z and tag ) r RunTagFilterr read_tensorsrrrrr'_process_datum) rr*r+r,rresponse_mappingrtf read_resultrdatads rr&zPrCurvesPlugin.pr_curves_implVs##D3%0))66 $ ,,** 7  KC??3+//4D| C!FJ$JT%8%8%;$J S ! K %KsB1ctj|j}tj|j}t j ||j ||dS)aA route (HTTP handler) that returns a response with tags. Returns: A response that contains a JSON object. The keys of the object are all the runs. Each run is mapped to a (potentially empty) dictionary whose keys are tags associated with run and whose values are metadata (dictionaries). The metadata dictionaries contain 2 keys: - displayName: For the display name used atop visualizations in TensorBoard. - description: The description that appears near visualizations upon the user hovering over a certain icon. r )rrr!r"rr% tags_impl)rr)r*r+s r tags_routezPrCurvesPlugin.tags_routewsR !!'//2 ..w?   T^^C46H  rc|jj||tj}|Dcic]}|i}}|j D]\}}|j D]~\}}tj |j } |jj| j||sM|jtj|jd|||<|Scc}w)zCreates the JSON object for the tags route response. Returns: The JSON object for the tags route response. )r"r1) displayName description)r list_tensorsrritemsparse_plugin_metadataplugin_contentrokversion display_namermarkdown_to_safe_htmlrA) rr*r+mappingrresulttag_to_time_seriesr time_seriesmds rr=zPrCurvesPlugin.tags_impls %%22 zx7K7K3 &--c#r'--'.}}  #C#$6$<$<$>  [33K4N4NO,,// CE#.#;#;#.#D#D#//$$s C   .s C(c4|j|jdS)zGets all routes offered by the plugin. Returns: A dictionary mapping URL path to route that handles it. )z/tagsz /pr_curves)r>r/rs rget_plugin_appszPrCurvesPlugin.get_plugin_appss__..  rcy)NFrPs r is_activezPrCurvesPlugin.is_activesrc0tjddS)Nztf-pr-curve-dashboardz PR Curves) element_nametab_name)rFrontendMetadatarPs rfrontend_metadataz PrCurvesPlugin.frontend_metadatas++0   rcd|j|j|j|jS)zConverts a TensorDatum into a dict that encapsulates information on it. Args: datum: The TensorDatum to convert. Returns: A JSON-able dictionary of PR curve data for 1 step. )_make_pr_entrystep wall_timenumpy)rdatums rr6zPrCurvesPlugin._process_datums%""5::u LLrc ntj}tj}tj}tj}|||gddfj t jd}t|dz } | dkDr|| dk(r| dz} | dkDr || dk(r| dz} |jd} tjdd| } ||D cgc] } t | }} ||D cgc] } t | }} ||D cgc] } t | }} ||D cgc] } t | }} |||tjd| fj|tjd| fj|d| |d| |d| |d| | d| jd Scc} wcc} wcc} wcc} w)a$Creates an entry for PR curve data. Each entry corresponds to 1 step. Args: step: The step. wall_time: The wall time. data_array: A numpy array of PR curve data stored in the summary format. Returns: A PR curve entry. Nr)axisgg?) r]r\ precisionrecalltrue_positivesfalse_positivestrue_negativesfalse_negatives thresholds)rTRUE_POSITIVES_INDEXFALSE_POSITIVES_INDEXTRUE_NEGATIVES_INDEXFALSE_NEGATIVES_INDEXastypeintsumlenshapenplinspacePRECISION_INDEXtolist RECALL_INDEX)rr\r] data_arraytp_indexfp_indextn_indexfn_index positivesend_index_inclusive end_indexnum_thresholdsrivrerfrgrhs rr[zPrCurvesPlugin._make_pr_entrys00110011(3Q67>>sCGGQGO !)nq0!A%)4G*HA*M 1 $ "A%)4G*HA*M'!+ #))!,[[c>: *4X*>?Q#a&??+5h+?@a3q6@@*4X*>?Q#a&??+5h+?@a3q6@@##((*9*4fh !6!6  !BCJJL,Zi8.z :,Zi8.z :$Zi0779  @@?@sF#*F(F- F2N)__name__ __module__ __qualname____doc__rrr1rrRequest applicationr/r&r>r=rQrTrYr6r[rSrrr r s{@&&K !!!"!F B!! " *.   M. rr )r^rswerkzeugr tensorboardrtensorboard.datartensorboard.backendrtensorboard.pluginsrtensorboard.plugins.pr_curverrTBPluginr rSrrrs8 #%)+1U [))U r