2Vh*ddlZddlZddlmZddlmZddlmZddlm Z ddl m Z ddl m Z  ddl Z edd gdd Zed d gdd Zed ddZedddZy#e$rdZ YKwxYw)N)logging) keras_export)legacy_h5_format) saving_lib) file_utils)io_utilszkeras.saving.save_modelzkeras.models.save_modelc |jdd}|jdd}|r[t|jdst|jdrtjd|nt d||r%t d t |jt|jdrtjd t|jd }|| } | xrtjj|}|r|stj|} | sy |r0t|jdrtj ||S|stj ||d St|jdrt#j$||||St d|d#t$rd}YwxYw)aSaves a model as a `.keras` file. Args: model: Keras model instance to be saved. filepath: `str` or `pathlib.Path` object. Path where to save the model. overwrite: Whether we should overwrite any existing model at the target location, or instead ask the user via an interactive prompt. zipped: Whether to save the model as a zipped `.keras` archive (default when saving locally), or as an unzipped directory (default when saving on the Hugging Face Hub). Example: ```python model = keras.Sequential( [ keras.layers.Dense(5, input_shape=(3,)), keras.layers.Softmax(), ], ) model.save("model.keras") loaded_model = keras.saving.load_model("model.keras") x = keras.random.uniform((10, 3)) assert np.allclose(model.predict(x), loaded_model.predict(x)) ``` Note that `model.save()` is an alias for `keras.saving.save_model()`. The saved `.keras` file is a `zip` archive that contains: - The model's configuration (architecture) - The model's weights - The model's optimizer's state (if any) Thus models can be reinstantiated in the exact same state. include_optimizerT save_formatF.h5.hdf5.keraszThe `save_format` argument is deprecated in Keras 3. We recommend removing this argument as it can be inferred from the file path. Received: save_format=zThe `save_format` argument is deprecated in Keras 3. Please remove this argument and pass a file path with either `.keras` or `.h5` extension.Received: save_format=z-The following argument(s) are not supported: aYou are saving your model as an HDF5 file via `model.save()` or `keras.saving.save_model(model)`. This file format is considered legacy. We recommend using instead the native Keras format, e.g. `model.save('my_model.keras')` or `keras.saving.save_model(model, 'my_model.keras')`. hf://N)zippedaInvalid filepath extension for saving. Please add either a `.keras` extension for the native Keras format (recommended) or a `.h5` extension. Use `model.export(filepath)` if you want to export a SavedModel for use with TFLite/TFServing/etc. Received: filepath=.)popstrendswithrwarning ValueErrorlistkeys startswithospathexists TypeErrorrask_to_proceed_with_overwriter save_modelrsave_model_to_hdf5) modelfilepath overwriterkwargsr r is_hfrproceeds K/home/dcms/DCMS/lib/python3.12/site-packages/keras/src/saving/saving_api.pyr r sL #6=**]E2K x= ! !"2 3s8}7M7M 8  OO)*5 7 )*5 7   ;FKKM"# %   8}./ C  M $ $W -E ~)9!9i88B  #h-((2$$UH55 $$UHUCC 8}./22 8Y(9     'Zq  *  s8$G GGzkeras.saving.load_modelzkeras.models.load_modelctt|jdxrtj|}t j |xr)t j t j|d}t|jd}t j|rt j |sx|sv|stt jtjtjj|}t j||tj|r|}d}|s|s|rtj ||||St|jdrt#j$|||St|jdrt'd|d t'd |d |d ) aLoads a model saved via `model.save()`. Args: filepath: `str` or `pathlib.Path` object, path to the saved model file. custom_objects: Optional dictionary mapping names (strings) to custom classes or functions to be considered during deserialization. compile: Boolean, whether to compile the model after loading. 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: A Keras model instance. If the original model was compiled, and the argument `compile=True` is set, then the returned model will be compiled. Otherwise, the model will be left uncompiled. Example: ```python model = keras.Sequential([ keras.layers.Dense(5, input_shape=(3,)), keras.layers.Softmax()]) model.save("model.keras") loaded_model = keras.saving.load_model("model.keras") x = np.random.random((10, 3)) assert np.allclose(model.predict(x), loaded_model.predict(x)) ``` Note that the model variables may have different name values (`var.name` property, e.g. `"dense_1/kernel:0"`) after being reloaded. It is recommended that you use layer attributes to access specific variables, e.g. `model.get_layer("dense_1").kernel`. rz config.jsonrT)custom_objectscompile safe_moder )r*r+zFile not found: filepath=z<. Please ensure the file is an accessible `.keras` zip file.$File format not supported: filepath=a. Keras 3 only supports V3 `.keras` files and legacy H5 format files (`.h5` extension). Note that the legacy SavedModel format is not supported by `load_model()` in Keras 3. In order to reload a TensorFlow SavedModel as an inference-only layer in Keras 3, use `keras.layers.TFSMLayer(za, call_endpoint='serving_default')` (note that your `call_endpoint` might have a different name).)rrzipfile is_zipfilerisdirrjoinris_remote_pathr get_temp_dirrrbasenamecopy load_modelrload_model_from_hdf5r)r#r*r+r, is_keras_zip is_keras_dirr& local_paths r(r6r6|sJx=))(38J8J9L##H-*2C2C-03L M $ $W -E !!(+  *__  # # %rww'7'7'A *-   j )!HL|u$$ )    8}./44 ^W   X   )'z2   28*='j, ,  zkeras.saving.save_weightsc ht|}||jdstd|||jdstd| tjj |}|r|stj|}|sytj|||fi|y#t $rd}YBwxYw)N .weights.h5z;The filename must end in `.weights.h5`. Received: filepath=)z weights.h5z weights.jsonz`The filename must end in `.weights.json` when `max_shard_size` is specified. Received: filepath=F) rrrrrrrrrrsave_weights_only)r"r#r$max_shard_sizer% filepath_strrr's r( save_weightsrAsx=Ll&;&;M&J "". 1    #L,A,A&- --9N <  )i88F   .KFK sB## B10B1zkeras.saving.load_weightsc t|}|jdr)|rtd|tj|||y|jds|jdr<|j dd}|rtd|tj||||y|jds|jd r|j d d }|rtd|t s td t j|d 5}d|jvr d|vr|d}|rtj|||ntj||dddytd|d#1swYyxYw)NrzInvalid keyword arguments: ) skip_mismatchr=z .weights.jsonobjects_to_skip)rCrDr rby_nameFz2Loading a H5 file requires `h5py` to be installed.r layer_names model_weightsr-zY. Keras 3 only supports V3 `.keras` and `.weights.h5` files, or legacy V1/V2 `.h5` files.) rrrrload_weights_onlyrh5py ImportErrorFileattrsr$load_weights_from_hdf5_group_by_nameload_weights_from_hdf5_group)r"r#rCr%r@rDrEfs r( load_weightsrQsx=LX& :6(CD D$$ 8=    } -1F1F2!**%6= :6(CD D$$  '+    u %)>)>w)G**Y. :6(CD DD YYx % HAGG+10Do& EEum!==aG H H28*=2 2   H Hs AE//E8)TN)NTT)F)rr.abslrkeras.src.api_exportrkeras.src.legacy.savingrkeras.src.savingrkeras.src.utilsrrrJrKr r6rArQr;r(rXs -4'&$ (*CDEfFfR(*CDE] F] @)*48L+L6)** +* Y DsA44A>=A>