o
    2hK
                     @   sP   d Z ddlmZ ddlmZ dZdZdZdZdZ	dZ
d	ZdZd
d Zdd ZdS )z0Internal information about the pr_curves plugin.    )summary_pb2)plugin_data_pb2	pr_curves               c                 C   s2   t jt|d}| }tj| |tjjt|ddS )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.
    )versionnum_thresholds)plugin_namecontent)display_namesummary_descriptionplugin_data)r   PrCurvePluginDataPROTO_VERSIONSerializeToStringr   SummaryMetadata
PluginDataPLUGIN_NAME)r   descriptionr   pr_curve_plugin_datar    r   `/var/www/html/chatgem/venv/lib/python3.10/site-packages/tensorboard/plugins/pr_curve/metadata.pycreate_summary_metadata$   s   r   c                 C   s0   t | ts	tdtj| }|jdkr|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.
    zContent type must be bytesr   )
isinstancebytes	TypeErrorr   r   
FromStringr
   )r   resultr   r   r   parse_plugin_metadata<   s   


r!   N)__doc__tensorboard.compat.protor   tensorboard.plugins.pr_curver   r   TRUE_POSITIVES_INDEXFALSE_POSITIVES_INDEXTRUE_NEGATIVES_INDEXFALSE_NEGATIVES_INDEXPRECISION_INDEXRECALL_INDEXr   r   r!   r   r   r   r   <module>   s   