我正在按照本教程对here中的结构化数据进行分类。我用我的数据（仅包含数字数据）对模型进行了训练。
我已经在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上找不到任何帮助。