keras - 如何在导入预训练的 keras 模型时修复“未知层：DenseFeatures”？

Question

我正在关注从这里对结构化数据进行分类的教程。我使用仅包含数字数据的数据训练了模型。

我已经在 google colab 上训练和优化了我的模型，并在本地下载了它以在一些新数据上进行测试。

但是，当我使用此代码段加载模型时。

from keras.models import load_model
model = load_model("my_model.h5")

我收到以下错误

Unknown layer: DenseFeatures

...连同痕迹。

我尝试custom_objects在加载模型时进行设置

model = load_model("my_model.h5", custom_objects={'DenseFeatures': tf.keras.layers.DenseFeatures})

但我仍然收到以下错误：

__init__() takes at least 2 arguments (3 given)

我可能做错了什么？我尝试浏览文档，但在 github 上找不到任何帮助。

标签： keraskeras-layer