o 2h@sJdZddlZddlmZddlmZddlmZGdddejjj Z dS)ztKeras integration for TensorBoard hparams. Use `tensorboard.plugins.hparams.api` to access this module's contents. N)api_pb2)summary) summary_v2c@s6eZdZdZd ddZddZd ddZd d d ZdS) CallbackzwCallback for logging hyperparameters to TensorBoard. NOTE: This callback only works in TensorFlow eager mode. NcCs\t||_||_tj|j|jd|durtdt|tr)tj j j ||_ dS||_ dS)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)selfwriterhparamsrr]/var/www/html/chatgem/venv/lib/python3.10/site-packages/tensorboard/plugins/hparams/_keras.py__init__"s   zCallback.__init__cCs(|jdur tdtstd|jS)Nz@hparams Keras callback cannot be reused across training sessionsz>hparams Keras callback only supported in TensorFlow eager mode)r RuntimeErrorrexecuting_eagerly)rrrr _get_writerHs zCallback._get_writercCsF~|tj|j|jdWddS1swYdS)Nr)r as_defaultrrr r )rlogsrrron_train_beginSs"zCallback.on_train_begincCsb~|ttj}|}tjj jj j |ddWdn1s'wYd|_ dS)Nr)step) rrrsession_end_pbrSTATUS_SUCCESSSerializeToStringrrr experimental write_raw_pbr)rrpbraw_pbrrr on_train_endXs  zCallback.on_train_end)N)__name__ __module__ __qualname____doc__rrrr(rrrrrs  & r) r, tensorflowrtensorboard.plugins.hparamsrrrkeras callbacksrrrrrs