h=SSKrSSKrSSKrSSKJr SSKrSSKJr SSK J r SSK r Sq SSK Jr \R "\5r"S S 5rg!\a Sq N'f=f) N)BytesIO) MessageToJson) _clean_tagT)ImageFc\rSrSrSS04SjrSr\S5r\SSj5r\SS j5r\SS j5r \SS j5r \SS j5r \SS j5r \SSj5r \SSj5r\SSj5r\S Sj5r\S!Sj5r\S"Sj5r\S5r\SSj5r\SSj5r\SSj5rSrg)# CometLoggerdisabledTcxSUlXlUSSLaSUlgUSSLa[SLa [S5egg)Nr TFzRComet and/or Python Image Library not installed. Run 'pip install comet-ml pillow')_logging _comet_configcomet_installed Exception)self comet_configs R/var/www/fran/franai/venv/lib/python3.13/site-packages/tensorboardX/comet_utils.py__init__CometLogger.__init__sL )  #t +!DM * % .?e3Kpq q4L .cF^[R"T5U4Sj5nU$)Nc>USnURc[(aSUlSUR;a[R"5 [R "5b[ RS5 [R"S0URD6Ul SUlURRSS5 URSLaT"U0UD6$g![an[ RU5 SnAN.wrappers7D}}$ %  & (:(:: 557C(78(0':':'PT=O=O'PD$$(DM$$..~~N}}$t.v..%!&NN1%%&sBC D!C;;D) functoolswraps)r'r(s` r_requiresCometCometLogger._requiresComets%   / ! /(rcbURR5 S[Rlg)zEnds an experiment.N)r"endrconfig experiment)rs rr/CometLogger.end6s" %)"rNcdUS:Xa [U5OUnURRXsXEU5 g)aLogs a general metric (i.e accuracy, f1).. Args: tag: String - Data identifier display_name: The title of the plot. If empty string is passed, `tag` will be used. value: Float/Integer/Boolean/String step: Optional. Used as the X axis when plotting on comet.ml epoch: Optional. Used as the X axis when plotting on comet.ml include_context: Optional. If set to True (the default), the current context will be logged along the metric. N)rr" log_metric)rtag display_namevaluestepepochinclude_contextnames rr5CometLogger.log_metric<s1#/""4z#, ##D$3 5rc<URRXX45 g)aLogs a key,value dictionary of metrics. Args: dic: key,value dictionary of metrics prefix: prefix added to metric name step: Optional. Used as the X axis when plotting on comet.ml epoch: Optional. Used as the X axis when plotting on comet.ml N)r" log_metrics)rdicprefixr9r:s rr?CometLogger.log_metricsOs $$S$>rc<URRXU5 g)zLogs a dictionary (or dictionary-like object) of multiple parameters. Args: parameters: key,value dictionary of parameters prefix: prefix added to metric name step: Optional. Used as the X axis when plotting on comet.ml N)r"log_parameters)r parametersrAr9s rrDCometLogger.log_parameters[s '' DArc BURRXUXEUU5 g)aLogs the audio Asset determined by audio data. Args: audio_data: String or a numpy array - either the file path of the file you want to log, or a numpy array given to scipy.io.wavfile.write for wav conversion. sample_rate: Integer - Optional. The sampling rate given to scipy.io.wavfile.write for creating the wav file. file_name: String - Optional. A custom file name to be displayed. If not provided, the filename from the audio_data argument will be used. metadata: Some additional data to attach to the the audio asset. Must be a JSON-encodable dict. overwrite: if True will overwrite all existing assets with the same name. copy_to_tmp: If audio_data is a numpy array, then this flag determines if the WAV file is first copied to a temporary file before upload. If copy_to_tmp is False, then it is sent directly to the cloud. step: Optional. Used to associate the audio asset to a specific step. N)r" log_audio)r audio_data sample_rate file_namemetadata overwrite copy_to_tmpr9s rrHCometLogger.log_audiofs$0 "":I#+ #' )rc<URRXU5 g)aLogs the text. These strings appear on the Text Tab in the Comet UI. Args: text: string to be stored step: Optional. Used to associate the asset to a specific step. metadata: Some additional data to attach to the the text. Must be a JSON-encodable dict. N)r"log_text)rtextr9rLs rrQCometLogger.log_texts !!$h7rc BURR"XUXE40UD6 g)aXLogs a histogram of values for a 3D chart as an asset for this experiment. Calling this method multiple times with the same name and incremented steps will add additional histograms to the 3D chart on Comet.ml. Args: values: a list, tuple, array (any shape) to summarize, or a Histogram object name: str (optional), name of summary step: Optional. Used as the Z axis when plotting on Comet.ml. epoch: Optional. Used as the Z axis when plotting on Comet.ml. metadata: Optional: Used for items like prefix for histogram name. kwargs: Optional. Additional keyword arguments for histogram. N)r"log_histogram_3d)rvaluesr<r9r:rLr%s r log_histogramCometLogger.log_histograms'" ))&*/ 4,2 4rctURSRn[U5nXS'URXQUS9 g)zLog Raw Histogram Data to Comet as an Asset. Args: tag: Name given to the logged asset summary: TensorboardX Summary protocol buffer with histogram data step: The Global Step for this experiment run. Defaults to None. rr<)datar<r9N)r8historlog_asset_data)rr6summaryr9histogram_protohistogram_raw_datas rlog_histogram_rawCometLogger.log_histogram_raws?"--*00*?;%(6" !3DIrcvURRXR5UR5XE5 g)zLog timeseries data. Args: name: (str) name of data x: array of x-axis values y: array of y-axis values overwrite: (optional, bool) if True, overwrite previous log step: (optional, int) the step value N)r" log_curvetolist)rr<xyrMr9s rrcCometLogger.log_curves( ""4QXXZQrc[U5n[R"U5n[U5nURR XVUS9 g)zLogs the image. Images are displayed on the Graphics tab on Comet.ml. Args: encoded_image_string: Required. An encoded image string tag: String - Data identifier step: Optional. Used to associate the image asset to a specific step. r9N)rropenrr" log_image)rencoded_image_stringr6r9buff image_pilr<s rlog_image_encodedCometLogger.log_image_encodeds?+,JJt$ # ""9">rc@URRXUXEU5 g)aLogs the Asset determined by file_data. Args: file_data: String or File-like - either the file path of the file you want to log, or a file-like asset. file_name: String - Optional. A custom file name to be displayed. If not provided the filename from the file_data argument will be used. overwrite: if True will overwrite all existing assets with the same name. copy_to_tmp: If file_data is a file-like object, then this flag determines if the file is first copied to a temporary file before upload. If copy_to_tmp is False, then it is sent directly to the cloud. step: Optional. Used to associate the asset to a specific step. N)r" log_asset)r file_datarKrMrNr9rLs rrrCometLogger.log_assets"$ ""9#.h @rc>URRXX4XV5 g)a:Logs the data given (str, binary, or JSON). Args: data: data to be saved as asset name: String, optional. A custom file name to be displayed If not provided the filename from the temporary saved file will be used. overwrite: Boolean, optional. Default False. If True will overwrite all existing assets with the same name. step: Optional. Used to associate the asset to a specific step. epoch: Optional. Used to associate the asset to a specific epoch. metadata: Optional. Some additional data to attach to the asset data. Must be a JSON-encodable dict. N)r"r\)rrZr<rMr9rLr:s rr\CometLogger.log_asset_datas" ''I(0 9rc Sn UcgUbIUR5R5R5nURSSn U SS:XaU SSn [ U5[ :Xa[ R"U5nO,UR5R5R5nURRXUXUUXxU 5 g)aZLog a multi-dimensional dataset and metadata for viewing with Comet's Embedding Projector (experimental). Args: vectors: the tensors to visualize in 3D labels: labels for each tensor image_data: (optional) list of arrays or Images image_preprocess_function: (optional) if image_data is an array, apply this function to each element first image_transparent_color: a (red, green, blue) tuple image_background_color_function: a function that takes an index, and returns a (red, green, blue) color tuple title: (optional) name of tensor template_filename: (optional) name of template JSON file Nrr) cpudetachnumpyshapetypelistnparrayr" log_embedding) rvectorslabels image_dataimage_preprocess_functionimage_transparent_colorimage_background_color_functiontitletemplate_filenamegroup image_sizes rrCometLogger.log_embeddings* >   !#)00288:J#))!"-J!}!'^ <4 XXf%FZZ\((*002F &&w '1'>'F',',  .rc0nXS'UR5US'UR5US'UR5US'XXS'XxS'SUS'[R"U5nURXUS 9 g ) asLogs a mesh as an asset Args: tag: Data identifier vertices: List of the 3D coordinates of vertices. colors: Colors for each vertex faces: Indices of vertices within each triangle. config_dict: Dictionary with ThreeJS classes names and configuration. step: step value to record walltime: Optional override default walltime (time.time()) seconds after epoch of event r6verticescolorsfaces config_dictwalltimemesh asset_typeriN)rdjsondumpsr\) rr6rrrrr9r mesh_jsons rlog_meshCometLogger.log_meshs} % ( 1 *$mmo ("\\^ '#.- (*"( ,JJy)  I6rc 0UnX%S'URXQUS9 g)zLogs a histogram as an asset. Args: tag: Data identifier asset_type: List of the 3D coordinates of vertices. step: step value to record rriN)r\)rr6rr9r% file_jsons rlog_raw_figureCometLogger.log_raw_figure8s$ ",, I6rc URSRnURRVs/sHofRPM nn[ R "UR[ RS9RU5n[U5V s/sH n SU- U -PM n n [S[ R"XRS55uppnnU U U UUUU US.nURUXS9 gs snfs sn f)Logs a Precision-Recall Curve Data as an asset. Args: tag: An identifier for the PR curve summary: TensorboardX Summary protocol buffer. step: step value to record r)dtype?c>UR5R5$Nflattenrdres r)CometLogger.log_pr_data..S!))+:L:L:Nr)TPFPTNFN precisionrecall thresholdsr<r<r9N)r8tensor tensor_shapedimsizer~fromiter float_valfloat32reshaperangemapvsplitr{r\)rr6r]num_thresholdsr9 tensor_protodr{rVirtpfptnfnrrpr_datas r log_pr_dataCometLogger.log_pr_dataEs}}Q'.. !-!:!:!>!>?!>A!>?\332::FNNuU8=n8MN8M1cN*Q.8M N,/0NPRPYPYZ`bnbnopbqPr,s) 6"$   G#9#@Os C5C:c [U5V s/sH n SU- U -PM n n [SUUUUUU/5upnnpgU UUUUUU U US. nURUXS9 gs sn f)rrc>UR5R5$rrrs rr-CometLogger.log_pr_raw_data..orr) rrrrrrrweightsr<rN)rrr\)rr6true_positive_countsfalse_positive_countstrue_negative_countsfalse_negative_countsrrrrr9rrrrrrrs rlog_pr_raw_dataCometLogger.log_pr_raw_databs9>n8MN8M1cN*Q.8M N,/0N ! !   Q-)B "$   G#9+OsA)rr"r )NNT)NNN)NN)NNNFTN)NNNNr)FN)NFTNN)NFNNN)NNNNzComet EmbeddingNN)__name__ __module__ __qualname____firstlineno__rr,r/r5r?rDrHrQrWr`rcrorrr\rrrrr__static_attributes__rrrr r s%/$6r0** DH#'55$ ? ?BB@D>B))6 8 8@D#44( J J R R ? ?=B8<@@(DH,099&8rs^ 5O   8 $o:o: Os A AA