keras
爱生活爱自己爱学习
这个作者很懒,什么都没留下…
展开
-
keras加载模型遇到ValueError: Unknown layer: TokenEmbedding
我使用bert进行了文本分类,想使用保存的模型对一些新的数据进行预测 发现会遇到错误ValueError: Unknown layer: TokenEmbedding。 万般搜索不得其解,在网上看到了一个方案https://blog.csdn.net/yscoder/article/details/103214078,尝试之后成功。 ...原创 2019-12-31 15:45:27 · 2960 阅读 · 0 评论 -
keras模型保存与加载
本文内容来自于个人理解和网上参考内容的综合,难免有错误(如能指正 不胜感激),仅供参考。 参考https://blog.csdn.net/tszupup/article/details/85198949 https://blog.csdn.net/zengNLP/article/details/94589469 1、模型保存及加载 1.1同时保存模型及参数 通过下面的方式加载 #...原创 2019-12-16 10:02:36 · 255 阅读 · 0 评论