U VcK @sPdZddlmZddlmZdZdZdZdZdZ dZ d Z dZ d d Z d d ZdS)z0Internal information about the pr_curves plugin.) summary_pb2)plugin_data_pb2Z pr_curvescCs2tjt|d}|}tj||tjjt|ddS)a[Create a `summary_pb2.SummaryMetadata` proto for pr_curves plugin data. Arguments: display_name: The display name used in TensorBoard. description: The description to show in TensorBoard. num_thresholds: The number of thresholds to use for PR curves. Returns: A `summary_pb2.SummaryMetadata` protobuf object. )versionnum_thresholds)Z plugin_namecontent) display_nameZsummary_descriptionZ plugin_data)rPrCurvePluginData PROTO_VERSIONZSerializeToStringrZSummaryMetadataZ PluginData PLUGIN_NAME)r descriptionr Zpr_curve_plugin_datar rI/tmp/pip-unpacked-wheel-g8kmtpbc/tensorboard/plugins/pr_curve/metadata.pycreate_summary_metadata$s rcCs0t|tstdtj|}|jdkr,|S|S)zParse summary metadata to a Python object. Arguments: content: The `content` field of a `SummaryMetadata` proto corresponding to the pr_curves plugin. Returns: A `PrCurvesPlugin` protobuf object. zContent type must be bytesr) isinstancebytes TypeErrorrr Z FromStringr )r resultrrrparse_plugin_metadata<s   rN)__doc__Ztensorboard.compat.protorZtensorboard.plugins.pr_curverrZTRUE_POSITIVES_INDEXZFALSE_POSITIVES_INDEXZTRUE_NEGATIVES_INDEXZFALSE_NEGATIVES_INDEXZPRECISION_INDEXZ RECALL_INDEXrrrrrrrs