Я следовал учебному пособию для классификации структурированных данных из здесь .Я обучил модель, как описано, с моими данными, которые состояли только из числовых данных.
Я обучил и оптимизировал свою модель на Google Colab и загрузил ее локально, чтобы протестировать ее для некоторых новых данных.
Однако, когда я загружаю модель с использованием этого фрагмента.
from keras.models import load_model
model = load_model("my_model.h5")
Я получаю следующую ошибку
Unknown layer: DenseFeatures
... вместе с трассировкой.
Я пытался настроить custom_objects при загрузке модели, какthis
model = load_model("my_model.h5", custom_objects={'DenseFeatures': tf.keras.layers.DenseFeatures})
Но я все еще получаю следующую ошибку:
__init__() takes at least 2 arguments (3 given)
Что я могу сделать не так?Я попытался просмотреть документацию, но не смог найти ничего полезного на github.