1VhdZddlZddlmZddlmZddlmZGddejjjZ y)ztKeras integration for TensorBoard hparams. Use `tensorboard.plugins.hparams.api` to access this module's contents. N)api_pb2)summary) summary_v2c.eZdZdZddZdZddZddZy)CallbackzwCallback for logging hyperparameters to TensorBoard. NOTE: This callback only works in TensorFlow eager mode. NcBt||_||_tj|j|j| t dt |tr9tjjjj||_ y||_ y)a'Create a callback for logging hyperparameters to TensorBoard. As with the standard `tf.keras.callbacks.TensorBoard` class, each callback object is valid for only one call to `model.fit`. Args: writer: The `SummaryWriter` object to which hparams should be written, or a logdir (as a `str`) to be passed to `tf.summary.create_file_writer` to create such a writer. hparams: A `dict` mapping hyperparameters to the values used in this session. Keys should be the names of `HParam` objects used in an experiment, or the `HParam` objects themselves. Values should be Python `bool`, `int`, `float`, or `string` values, depending on the type of the hyperparameter. trial_id: An optional `str` ID for the set of hyperparameter values used in this trial. Defaults to a hash of the hyperparameters. Raises: ValueError: If two entries in `hparams` share the same hyperparameter name. trial_idNz3writer must be a `SummaryWriter` or `str`, not None)dict_hparams _trial_idr hparams_pb TypeError isinstancestrtfcompatv2rcreate_file_writer_writer)selfwriterhparamsr s R/home/dcms/DCMS/lib/python3.12/site-packages/tensorboard/plugins/hparams/_keras.py__init__zCallback.__init__"sv4W  !dmmdnnE >E  $99<<//BB6JDL!DLc|j tdtjs td|jS)Nz@hparams Keras callback cannot be reused across training sessionsz>hparams Keras callback only supported in TensorFlow eager mode)r RuntimeErrorrexecuting_eagerly)rs r _get_writerzCallback._get_writerHsF << R ##%P ||rc~|jj5tj|j|j dddy#1swYyxYw)Nr )r as_defaultrrr r )rlogss ron_train_beginzCallback.on_train_beginSsK      * * , G   t}}t~~ F G G Gs ,AAcl~|jj5tjtj }|j }tjjjjj|ddddd|_ y#1swYd|_ yxYw)Nr)step) r r"rsession_end_pbrSTATUS_SUCCESSSerializeToStringrrr experimental write_raw_pbr)rr#pbraw_pbs r on_train_endzCallback.on_train_endXs      * * , K''(>(>?B))+F IILL - - : :6 : J K  K s A3B##B3)N)__name__ __module__ __qualname____doc__rr r$r.rrrrs $"L G rr) r2 tensorflowrtensorboard.plugins.hparamsrrrkeras callbacksrr3rrr8s6 //2Brxx!!**Br