2VhrtdZddlZddlZddlZddlZddlZddlmZddlm Z ddl m Z ddl m Z ddlmZddlmZdd lmZeeeefZd ZGd d ZGd dZe ddZdZGddZdZ dZ!dZ"e ddgdZ#dZ$d#dZ%d#dZ&dZ'dZ(e dd g d$d!Z) d#d"Z*y)%z6Object config serialization and deserialization logic.N) api_export)backend) keras_export) global_state)object_registration) python_utils) tensorflow) activations constraints initializerslossesmetrics optimizers regularizersceZdZdZdZy)SerializableDictc ||_yN)config)selfrs R/home/dcms/DCMS/lib/python3.12/site-packages/keras/src/saving/serialization_lib.py__init__zSerializableDict.__init__!s  c,t|jSr)serialize_keras_objectrrs r serializezSerializableDict.serialize$s%dkk22rN)__name__ __module__ __qualname__rrrrrr s 3rrc$eZdZdZddZdZdZy) SafeModeScopezBScope to propagate safe mode flag to nested deserialization calls.c||_yr) safe_mode)rr%s rrzSafeModeScope.__init__+s "rcbt|_tjd|jyNsafe_mode_saving) in_safe_modeoriginal_valuerset_global_attributer%rs r __enter__zSafeModeScope.__enter__.s!*n))*s  , ,-? @@rceZdZdZdZdZy)ObjectSharingScopez?Scope to enable detection and reuse of previously seen objects.c\tjditjdiyNshared_objects/id_to_obj_mapshared_objects/id_to_config_mapr4rs rr,zObjectSharingScope.__enter__Es$))*H"M))*KRPrc\tjddtjddyr<r4r.s rr1zObjectSharingScope.__exit__Is&))*H$O)) -t rN)rrr r2r,r1r!rrr:r:BsIQ rr:cVtjd}||j|dSy)z:Retrieve an object previously seen during deserialization.r=N)rr8get)obj_id id_to_obj_maps rget_shared_objectrDPs5 55&M   ..!rc|ddk(rd|d<tjd}|ytt|}||vr|||<y||d<||}||d<y)z>Call after serializing an object, to keep track of its config.module__main__Nr>shared_object_id)rr8intid)objrid_to_config_maprB prev_configs r!record_object_after_serializationrNYsy h:%x#88) C\F %%#) %+!"&v. *0 &'rc>tjd}|y|||<y)zFCall after deserializing an object, to keep track of it in the future.r=Nr7)rKrBrCs r#record_object_after_deserializationrPks* 55&MM&rz#keras.saving.serialize_keras_objectz"keras.utils.serialize_keras_objectc ||St|tr|St|ttfr5|Dcgc] }t |}}t|tr t|S|St|t r t |St|trdd|jdidSt|trBdt |jt |jt |jddSt|ttrdidSt|tj rIt#|d d}|rt|}|d j$|d <d |j&|j(|d dSt*j,r8t|t*j.r|j0|j3SdStj4|rFd tj6|j9tj:|j(ddSt|j<t>j@k(rlt|t>jBrB|jDd kDr3d|j9tj:|j(ddS|jGSt*j,r&t|t*jHr |j$St|tJjLrV|j@dk(rGtOjPdtSjT|dddtWjX|idSt*j,rpt|t*jZrV|j]}tt_d|}d|j`j@|j`j<|ddStc|}te|j`|}|tg||ti|||St|tJjLr |j<}n|j`j<}|j`j@}|dk(rd} nOt|tJjLrtkjl|} ntkjl|j`} |||| d} tg|| ti|| | Scc}w)a Retrieve the config dict by serializing the Keras object. `serialize_keras_object()` serializes a Keras object to a python dictionary that represents the object, and is a reciprocal function of `deserialize_keras_object()`. See `deserialize_keras_object()` for more information about the config format. Args: obj: the Keras object to serialize. Returns: A python dict that represents the object. The python dict can be deserialized via `deserialize_keras_object()`. N __bytes__valueutf-8) class_namer __slice__)startstopstep __ellipsis___keras_historyr__keras_tensor__)shapedtype keras_history __tensor__)rSr^ __numpy__zaThe object being serialized includes a `lambda`. This is unsafe. In order to reload the object, you will have to pass `safe_mode=False` to the loading function. Please avoid using `lambda` in the future, and use named Python functions instead. This is the `lambda` being serialized: ) stacklevel __lambda__ct|tjr|jSt|tjr |j S|Sr) isinstancetf TensorShapeas_listDTypenamexs rz(serialize_keras_object..sG!!R^^4IIK%/q"(($;!&&BCr __typespec__)rU spec_namerFrregistered_namebuiltinsrFrUrrq)7rf PLAIN_TYPESlisttuplerdictserialize_dictbytesdecodeslicerWrXrYtypeEllipsisr KerasTensorgetattrrkr]r^rg availablerh_dimsri is_tensorconvert_to_numpytoliststandardize_dtypernprndarrayndimitemrjtypes FunctionTypewarningswarninspect getsourcer func_dumpTypeSpec _serializemap __class___get_class_or_fn_configserialize_with_public_classget_build_and_compile_configrNrget_registered_name) rKrm config_arrhistory ts_config inner_configconfig_with_public_classrFrUrqrs rrrus+* { #{# #e}%9<=A,Q/= =$.sE$:uZ J J#tc""#u% 7 34  #u%/ :.sxx8.sxx8  #tH~&,;;#w**+#/6 7mG GAJ,!(   || 37 # 5s{{}?4?& 11#6==? 22399=   Cyr{{* c2:: &388a<) ZZ\$66syyA 88:  || 31xx#u))*s||z/I  6 7>6G6G6L5M  O  '//4   || 3 4NN$       )//mm..#   +3/L: |  +$S*BC)#/GH''#u))*))''J  c5-- .1EEcJO1EE O  * F !f-%c62 MM>sSct|dr |j}|t||d<t|dr |j}|t||d<y)Nget_build_config build_configget_compile_configcompile_config)hasattrrrxr)rKrrrs rrrsbs&'++-  #%3L%AF> "s()//1  %'5n'EF# $ rctj|}|3tj|}|y|j|j ||dS|j d}dj|dd|d|ddS)zSerializes classes from public Keras API or object registration. Called to check and retrieve the config of any class that has a public Keras API or has been registered as serializable via `keras.saving.register_keras_serializable()`. Nrs.)rget_name_from_symbolrrrrsplitjoin)clsrkeras_api_namerqpartss rrr&s 44S9N-AA#F  "nn,,".      %E((5":&Bi  rc|r|d||dStj|}|r*|jd}dj|ddd||dSt j |}|s|j dk(sy|j d||dS)aOSerializes functions from public Keras API or object registration. Called to check and retrieve the config of any function that has a public Keras API or has been registered as serializable via `keras.saving.register_keras_serializable()`. If function's module name is already known, returns corresponding config. functionrsrNrrr)rrrrrrr)fnrfn_module_namerrrqs rserialize_with_public_fnrIs$$%    44R8N$$S)hhuSbz*$%   .AA"Er}} 'Bmm$.   rcht|tjrtj|St |dr<|j }t|tstd|d|t|St |drtj|Std|dt|d)z1Return the object's config depending on its type. get_configzThe `get_config()` method of z$ should return a dict. It returned: rzCannot serialize object z of type zG. To be serializable, a class must implement the `get_config()` method.) rfrrrrrrrw TypeErrorrxr|)rKrs rrrms#u))*"66s;;sL!!&$'/u5((.x1 f%% j !"66s;;&se9T#YK@@ @  rch|jDcic]\}}|t|c}}Scc}}wr)itemsr)rKkeyrSs rrxrxs+AD M:3C'. . MM Ms.z%keras.saving.deserialize_keras_objectz$keras.utils.deserialize_keras_objectc z t}||n|}|jdd}|xsi}tjdi}tj }i|||}|yt |tr|r|j|||St |ttfr|Dcgc]}t|||c}S| d\} } } t |trd|vr|d} d|vrtd||r1|d|vs'|jd |vst | tr| |vrd } nM|dd k(r|d } | d k(r|d}n5|d }n/|jd dtd|dd|dd|d}| sY||vr|St ||tj rtt#|||| |Stt%||| |St |t&r|St |tstd|d|vsd|vr0|j)D cic]\} } | t| ||c} } S|d}|dxsi} |xsi}|dk(r)t+j,| d| d}| d|_|S|dk(rt+j0| d| dS|dk(rt3j4| d| dS|ddk(r| dj7dS|ddk(rt8S|dd k(r:t;t| d!||t| d"||t| d#||S|dd$k(r%|r td%t=j>| dSt@I|dd&k(rAtC|d'|d |d d(||)}tEd*| } |jGt| S|jd d}|jd |}|d k(r| }tC|||d ||)Sd+|vrtI|d+}||StC|||d(||)}t |tj r|StK|d,std-|d.|t jL|}tO|}|5|5 |jQ| }|jd1d}|r$|jRs|jU|d |_)|jd2d}|r|jW|d |_,ddddddd+|vrt[|d+Scc}wcc} } w#t$r}t|d/|d0|d}~wwxYw#1swYUxYw#1swYYxYw)3a:Retrieve the object by deserializing the config dict. The config dict is a Python dictionary that consists of a set of key-value pairs, and represents a Keras object, such as an `Optimizer`, `Layer`, `Metrics`, etc. The saving and loading library uses the following keys to record information of a Keras object: - `class_name`: String. This is the name of the class, as exactly defined in the source code, such as "LossesContainer". - `config`: Dict. Library-defined or user-defined key-value pairs that store the configuration of the object, as obtained by `object.get_config()`. - `module`: String. The path of the python module. Built-in Keras classes expect to have prefix `keras`. - `registered_name`: String. The key the class is registered under via `keras.saving.register_keras_serializable(package, name)` API. The key has the format of '{package}>{name}', where `package` and `name` are the arguments passed to `register_keras_serializable()`. If `name` is not provided, it uses the class name. If `registered_name` successfully resolves to a class (that was registered), the `class_name` and `config` values in the dict will not be used. `registered_name` is only used for non-built-in classes. For example, the following dictionary represents the built-in Adam optimizer with the relevant config: ```python dict_structure = { "class_name": "Adam", "config": { "amsgrad": false, "beta_1": 0.8999999761581421, "beta_2": 0.9990000128746033, "decay": 0.0, "epsilon": 1e-07, "learning_rate": 0.0010000000474974513, "name": "Adam" }, "module": "keras.optimizers", "registered_name": None } # Returns an `Adam` instance identical to the original one. deserialize_keras_object(dict_structure) ``` If the class does not have an exported Keras namespace, the library tracks it by its `module` and `class_name`. For example: ```python dict_structure = { "class_name": "MetricsList", "config": { ... }, "module": "keras.trainers.compile_utils", "registered_name": "MetricsList" } # Returns a `MetricsList` instance identical to the original one. deserialize_keras_object(dict_structure) ``` And the following dictionary represents a user-customized `MeanSquaredError` loss: ```python @keras.saving.register_keras_serializable(package='my_package') class ModifiedMeanSquaredError(keras.losses.MeanSquaredError): ... dict_structure = { "class_name": "ModifiedMeanSquaredError", "config": { "fn": "mean_squared_error", "name": "mean_squared_error", "reduction": "auto" }, "registered_name": "my_package>ModifiedMeanSquaredError" } # Returns the `ModifiedMeanSquaredError` object deserialize_keras_object(dict_structure) ``` Args: config: Python dict describing the object. custom_objects: Python dict containing a mapping between custom object names the corresponding classes or functions. safe_mode: Boolean, whether to disallow unsafe `lambda` deserialization. When `safe_mode=False`, loading an object has the potential to trigger arbitrary code execution. This argument is only applicable to the Keras v3 model format. Defaults to `True`. Returns: The object described by the `config` dictionary. Nmodule_objectscustom_objects_scope_dict)custom_objectsr%)NNFrrUz%Unknown `config` as a `dict`, config=rqTrrFrr_z#Cannot deserialize object of type `z`. If `zj` is a custom class, please register it using the `@keras.saving.register_keras_serializable()` decorator.r)rzCould not parse config: r\r]r^)r^r_r`rSrarRrTrZrVrWrXrYrdaRRequested the deserialization of a `lambda` object. This carries a potential risk of arbitrary code execution and thus it is disallowed by default. If you trust the source of the saved model, you can pass `safe_mode=False` to the loading function in order to allow `lambda` loading, or call `keras.config.enable_unsafe_deserialization()`.rorpclass)obj_type full_configrct|trtj|St tj t |rtt|S|Sr)rfrurgrhrdtypesstrrrls rrnz*deserialize_keras_object..sLa&q!)0 3q6(Bgb!nIJrrH from_configz&Unable to reconstruct an instance of 'zM' because the class is missing a `from_config()` method. Full object config: z could not be deserialized properly. Please ensure that components that are Python object instances (layers, models, etc.) returned by `get_config()` are explicitly deserialized in the model's `from_config()` method. config=z. Exception encountered: rr).r)poprr8rGLOBAL_CUSTOM_OBJECTSrfrrArurvdeserialize_keras_objectrw ValueErrorrrrrrrtrrr~ _pre_serialization_keras_historyconvert_to_tensorrarrayencoder}r{r func_loadrg_retrieve_class_or_fnr _deserializerDrCustomObjectScoper#rbuiltbuild_from_configcompile_from_configcompiledrP)rrr%r0safe_scope_argrtlcogcormrrhas_custom_objectrrSrUrKrFrqfn_namercustom_obj_scopesafe_mode_scopeinstanceerrs rrrsP"^N"0"<)IZZ 0$7N#)rN  , ,-H" MD  3 3C66$6#6N ~ 63    v & 2f%%&4-(    %.I   !:K7 n&7 fd #6!%h/ 6) ;F8D |$6::/0NB|S1$6%)! %3!'!1!Z/#H-F#$56F::h,4#"<012"<012%% - ^+ .0%2D2DE/,&v.$2  ,+"6* .  &+& fd #26(;<<6!XV%; %lln  U )n     %J(#)rL#)rN''!!  !g)> 0  [ xx W-\'5JKK l{*G$++G44 l~- l{* $W%-#  %V$-#  %V$-#   "l|+ J %%l7&;<< ~&..@# ;  $ % 8 )      l 344ZZ$ 'Fjj!2J?OZ$   )   V#'9 :; ?J %  C#u))* 3 &4ZLA##)( ,  +<<^L#I.O %?% |4Hzz.$7   & &| 4!HN$4d;   ( ( 8 $H '%%*V#+ f/0  O} V z % %X%A! F  %%%%sUS6S;T1T%T-A$T%T1 T" TT""T%%T. *T11T:c |dk(rtj||}ntj||}||S|r|dk(s|jdr!|dz|z}tj|}||S|dk(rj|dk(ret D]&} tjd| zdz|z}|$|cSd|vr2d|z} |j D]\} } | j| s| cS|jdd d } | d vr6 tj|} t| j|d}||S td|d |d|#t$rtd |d |d|d|wxYw)Nrrkeraszkeras.rrr>)maxsplitr>rkeras_cv keras_hub keras_nlpzCould not deserialize z 'z' because its parent module z) cannot be imported. Full object config: zCould not locate zr'. Make sure custom classes are decorated with `@keras.saving.register_keras_serializable()`. Full object config: )rget_registered_object startswithrget_symbol_from_nameBUILTIN_MODULESrendswithr importlib import_modulevarsrAModuleNotFoundErrorr)rkrqrFrrr custom_objapi_namerKmodseparated_name custom_name custom_objectpackages rrrs :(>>  )>> N  W  1 1( ;|d*H11(;C z !f &:&  55sNS(4/?J  $!$t2@2F2F2H-.K"++N;,,- ,,sQ,/2 E E --f53immD$/?J#  H:Rv. +m - ' ,XJb?))/1++6-9 s <3E!E&r)NT)+r2rrrrnumpyr keras.srcrrkeras.src.api_exportrkeras.src.backend.commonrkeras.src.savingrkeras.src.utilsrkeras.src.utils.module_utilsr rgrrIfloatboolrtrrr#r5r)r:rDrNrPrrrrrrxrrr!rrrs <  -10(9C% 33   :;A<A A   /1$ -, \  \~   F! H 0N/.,0V  Vt JNGr