a Sic/5@sdZddlZddlZddlmmZddlZddlmZddlm Z ddlm Z ddl m Z ddl mZddlmZdd lmZdd lmZdd lmZd d Zd(ddZddZd)ddZd*ddZddZd+ddZddZddZd d!Z d"d#Z!d$d%Z"d&d'Z#dS),z$Utils related to keras model saving.N)backend)losses) optimizers)base_layer_utils) optimizer_v1) generic_utils) version_utils)ask_to_proceed_with_overwrite) tf_loggingcCs t|ddrdd|jDSdS)aPConvert metrics from a Keras model `compile` API to dictionary. This is used for converting Keras models to Estimators and SavedModels. Args: model: A `tf.keras.Model` object. Returns: Dictionary mapping metric names to metric instances. May return `None` if the model does not contain any metrics. _compile_metricsNcSsi|] }|j|qS)name).0mr r U/var/www/html/django/DPS/env/lib/python3.9/site-packages/keras/saving/saving_utils.py 4z)extract_model_metrics..)getattr_compile_metric_functionsmodelr r rextract_model_metrics$s rFcCs&|j| d}|durdSt|}|S)atInspect model to get its input signature. The model's input signature is a list with a single (possibly-nested) object. This is due to the Keras-enforced restriction that tensor inputs must be passed in as the first argument. For example, a model with input {'feature1': , 'feature2': } will have input signature: [{'feature1': TensorSpec, 'feature2': TensorSpec}] Args: model: Keras Model object. keep_original_batch_size: A boolean indicating whether we want to keep using the original batch size or set it to None. Default is `False`, which means that the batch dim of the returned input signature will always be set to `None`. Returns: A tuple containing `(args, kwargs)` TensorSpecs of the model call function inputs. `kwargs` does not contain the `training` argument. ) dynamic_batchN)NN) save_spec_enforce_names_consistency)rkeep_original_batch_size input_specsr r rmodel_call_inputs8s rcCs2t|tjjrtd|dtd|ddS)NzModel z cannot be saved because the input shape is not available. Please specify an input shape either by calling `build(input_shape)` directly, or by calling the model on actual data using `Model()`, `Model.fit()`, or `Model.predict()`.a cannot be saved either because the input shape is not available or because the forward pass of the model is not defined.To define a forward pass, please override `Model.call()`. To specify an input shape, either call `build(input_shape)` directly, or call the model on actual data using `Model()`, `Model.fit()`, or `Model.predict()`. If you have a custom training step, please make sure to invoke the forward pass in train step through `Model.__call__`, i.e. `model(inputs)`, as opposed to `model.call()`.) isinstancekerasmodels Sequential ValueErrorrr r rraise_model_input_errorVs   r#csn|dur"tjtjjjr"jj}|r0|}i}nt\}}|durLttjfdd}|j |i|S)aTrace the model call to create a tf.function for exporting a Keras model. Args: model: A Keras model. input_signature: optional, a list of tf.TensorSpec objects specifying the inputs to the model. Returns: A tf.function wrapping the model's call function with input signatures set. Raises: ValueError: if input signature cannot be inferred from the model. Ncsjjdd||dd\}}tjddddd|i|}Wdn1sV0Yj}|durddlm}||}t j |}d d t ||DS) z._wrapped_model..) _call_spec set_arg_valuer call_contextenter output_names keras.enginer)create_pseudo_output_namestfnestflattenzip)argskwargsoutputsr/r)rr r_wrapped_models  ,   z(trace_model_call.._wrapped_model) rcallr2 __internal__functionFunctioninput_signaturerr#get_concrete_function)rr>Z model_args model_kwargsr9r rrtrace_model_callms rATc Csddlm}ddlm}d|jji}z||d<Wn,ty`}z|rL|WYd}~n d}~00tt |t |d}|j r |r t |j t jrtdnj|jr |jd d }|d dt||d <t |j |jrtd nt|j j|j d} | |d d<|S)z3Returns a dictionary containing the model metadata.r) __version__) optimizer_v2 class_nameconfigN) keras_versionr model_configa<TensorFlow optimizers do not make it possible to access optimizer attributes or optimizer state after instantiation. As a result, we cannot save the optimizer as part of the model save file. You will have to compile your model again after loading it. Prefer using a Keras optimizer instead (see keras.io/optimizers).F) user_metrics optimizertraining_configzOptimizers loaded from a SavedModel cannot be saved. If you are calling `model.save` or `tf.keras.models.save_model`, please set the `include_optimizer` option to `False`. For `tf.saved_model.save`, delete the optimizer from the model.)rDrEoptimizer_config)rrBkeras.optimizers.optimizer_v2rC __class____name__ get_configNotImplementedErrordictstrrrIrr TFOptimizerloggingwarning_compile_was_called_get_compile_argspop_serialize_nested_configRestoredOptimizerrget_registered_name) rinclude_optimizerrequire_configrFrCrGemetadatarJrKr r rmodel_metadatasF        r`cCs|stj|rt|SdS)z3Returns whether the filepath should be overwritten.T)ospathisfiler )filepath overwriter r rshould_overwritesrfc Cs|dur i}t||d}t|}d}|dd}|durNttj|}d}|dd}|durptt|}d}|dd} | durtt| }t|dr|dnd} |d} Wdn1s0Yt ||||| | dS) z4Return model.compile arguments from training config.NrKlossmetricsweighted_metricssample_weight_mode loss_weights)rIrgrhrirkrj) rCustomObjectScoper deserializeget_deserialize_nested_configr_deserialize_metrichasattrrQ) rJcustom_objectsrKrIrg loss_configrhmetrics_configriweighted_metrics_configrjrkr r r!compile_args_from_training_configsD       &rvcsxdd}|durdS||r$|St|trDfdd|DSt|ttfrdfdd|DStd|d dS) z=Deserializes arbitrary Keras `config` using `deserialize_fn`.cSs(t|trd|vrdSt|tr$dSdS)NrDTF)rrQrRobjr r r_is_single_objects  z5_deserialize_nested_config.._is_single_objectNcsi|]\}}|t|qSr ro)rkvdeserialize_fnr rr sz._deserialize_nested_config..csg|]}t|qSr rz)rrxr}r r %sz._deserialize_nested_config..zrSaved configuration not understood. Configuration should be a dictionary, string, tuple or list. Received: config=.)rrQitemstuplelistr")r~rEryr r}rros$   rocCsdd}tj||S)z/Serialized a nested structure of Keras objects.cSst|rt|S|SN)callablerserialize_keras_objectrwr r r _serialize_fn2s z/_serialize_nested_config.._serialize_fn)r2r3 map_structure)rErr r rrY/srYcCs"ddlm}|dvr|S||S)z7Deserialize metrics, leaving special strings untouched.r)rh)accuracyacc crossentropyce)rrhrm) metric_configmetrics_moduler r rrp:s rpcs`dddd}tj|}tfdd|DoHtfdd|D }|r\tj||}|S)z5Enforces that either all specs have names or none do.cSs|dupt|do|jduSNr )rqr specr r r _has_nameIsz-_enforce_names_consistency.._has_namecSst|}t|drd|_|Sr)copydeepcopyrq_namerr r r _clear_nameLs  z/_enforce_names_consistency.._clear_namec3s|]}|VqdSrr )rsrr r Srz-_enforce_names_consistency..)r2r3r4anyallr)specsr flat_specsname_inconsistencyr rrrFs " rcCsdt|s`|jdur`z4|jjs,|j|j|jjsF|j|j|jWntdYn0dS)NzCompiled the loaded model, but the compiled metrics have yet to be built. `model.compile_metrics` will be empty until you train or evaluate the model.) ris_v1_layer_or_modelr8 compiled_lossbuiltbuildcompiled_metricsrTrUrr r rtry_build_compiled_arguments\srcCs|dp|dp|dS)Nz.h5z.kerasz.hdf5)endswith)rdr r ris_hdf5_filepathns  r)F)N)TT)N)$__doc__rratensorflow.compat.v2compatv2r2rrrrr0rZkeras.optimizersr keras.utilsrrZkeras.utils.io_utilsr tensorflow.python.platformr rTrrr#rAr`rfrvrorYrprrrr r r rs4           4 8 0