&`i ddlmZddlZddlmZddlmZddlm Z dZ GddZ Gd d Z e d Gd deZ dS))DictN)TensorflowCheckpoint)RayReportCallback) PublicAPIzxThe `ray.tune.integration.keras` module is deprecated in favor of `ray.train.tensorflow.keras.ReportCheckpointCallback`.ceZdZdZdZdS)TuneReportCallbackYDeprecated. Use :class:`ray.train.tensorflow.keras.ReportCheckpointCallback` instead.c*ttNDeprecationWarning_DEPRECATION_MESSAGEclsargskwargss n/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/ray/tune/integration/keras.py__new__zTuneReportCallback.__new__ !5666N__name__ __module__ __qualname____doc__rrrrr/QQ77777rrceZdZdZdZdS)_TuneCheckpointCallbackr c*ttr r rs rrz_TuneCheckpointCallback.__new__rrNrrrrrrrrralpha) stabilityc.eZdZdZdedefdZdefdZdS)TuneReportCheckpointCallbackaWKeras callback for Ray Tune reporting and checkpointing. .. note:: Metrics are always reported with checkpoints, even if the event isn't specified in ``report_metrics_on``. Example: .. code-block:: python ############# Using it in Ray Tune ############### from ray.tune.integrations.keras import TuneReportCheckpointCallback def train_fn(): model = build_model() model.fit(dataset_shard, callbacks=[TuneReportCheckpointCallback()]) tuner = tune.Tuner(train_fn) results = tuner.fit() Args: metrics: Metrics to report. If this is a list, each item describes the metric key reported to Keras, and it's reported under the same name. If this is a dict, each key is the name reported and the respective value is the metric key reported to Keras. If this is None, all Keras logs are reported. report_metrics_on: When to report metrics. Must be one of the Keras event hooks (less the ``on_``), e.g. "train_start" or "predict_end". Defaults to "epoch_end". checkpoint_on: When to save checkpoints. Must be one of the Keras event hooks (less the ``on_``), e.g. "train_start" or "predict_end". Defaults to "epoch_end". metrics checkpointcHtj||dS)N)r&raytunereport)selfr%r&s r_save_and_report_checkpointz8TuneReportCheckpointCallback._save_and_report_checkpointBs" J77777rcDtj|dSr r()r,r%s r_report_metricsz,TuneReportCheckpointCallback._report_metricsGs      rN)rrrrrrr-r/rrrr$r$s^  D88)=8888 !t!!!!!!rr$)typingrray.tuner)ray.train.tensorflowrray.train.tensorflow.kerasrray.util.annotationsrrrrr$rrrr5s555555888888******= 7777777777777777 W)!)!)!)!)!#4)!)!)!)!)!r