無法加載保存在 keras / tensorflow 中的 model？

Question

我使用autokeras和TensorFlow 2.5 訓練了 model。
我使用 Keras (TensorFlow) 主頁上說明的兩種方法保存了預訓練的 model。 model.save(f'model_auto_keras{max_trials}.h5') model.save("keras_test_save_model")

當我想使用model = tf.keras.models.load_model(f'model_auto_keras{max_trials}.h5')和model1 = tf.keras.models.load_model("keras_test_save_model/") save_model 加載保存的 model 時，這兩種方法都不是在我的情況下做得很好。

說 ValueError: Unknown layer: Custom> ValueError

ValueError: Unknown layer: Custom>MultiCategoryEncoding.
Please ensure this object is passed to the `custom_objects` argument. See 
https://www.tensorflow.org/guide/keras/save_and_serialize#registering_the_custom_object for 
details.

主要問題是自定義層 >> MultiCategoryEncoding在 keras 中不可用。

運行時錯誤

Answer 1

@krishna 你可以試試：

model = tf.keras.models.load_model('model.h5', custom_objects={'CategoryLayerName': tf.keras.layers.CategoryEncoding()})

在您的 model 聲明中，使用層名稱作為 CategoryEncoding 層。

我不確定它應該是 tf.keras.layers.CategoryEncoding() 還是 tf.keras.layers.CategoryEncoding