- 博客(3)
- 资源 (1)
- 收藏
- 关注
原创 keras自定义层对自定义参数进行保存
问题描述 在使用keras的过程中,有时会在自定义层中定义自己的参数,然而调用 json_model = model.to_json() 后发现自定义的参数并没有保存到模型中,造成模型加载时报错。 解决方案 需要在该层中实现 get_config(self) 方法,将自定义的参数保存到一个map中。聚义可以参考本人的《Keras 实现对bert编码的封装实现》中的具体写法。 ...
2019-05-19 23:51:51 2916 2
原创 Keras 实现对bert编码的封装实现
bert编码突袭了NLP各大榜单,为了让大家更方便的在自己的数据集合上测试bert的威力,在此分享一下,keras版本的nert编码层。直接上代码了。 class b_embeding_layer_b(keras.layers.Layer): """自定义层""" def __init__(self, max_seq_len=50,model_dir = r"F:\glove.6...
2019-05-19 23:39:29 1904
原创 Bert编码训练NER实体偏移的问题
在这里记录一下踩过的坑 在训练样本中几乎没有前面一大串英文或者数字,然后后面跟实体的情况,所以训练起来还挺平稳的。 但是在预测的时候遇到了这种情况,造成实体可以识别出来,但是实体的预测结果下标错了(表现为下标提前了) 造成这一问题的原因是在给bert传文本得到文本特征的时候,可以穿字符串的文本,也可以自己token完成后传一个token后的list。 如果传递的是字符串,默认系统的token,对于...
2019-05-19 23:30:32 1414 5
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人