保存机器学习 keras 模型时出现问题

Question

我按照这个关于 keras 的教程 https://keras.io/examples/nlp/semantic_similarity_with_bert/

我想用这个命令保存模型

model.save("saved_model/my_model")

我在保存模型时收到此警告 enter image description here

然后当我想加载模型以使用此命令时

 tf.keras.models.load_model('saved_model/my_model')

我收到了这个错误 enter image description here

这是保存模型的好方法吗？

Answer 1

您的第一个结构在字典中。您必须从 dict 中提取项目才能摆脱您的错误。尝试检查this。