用keras调用bert比较好用。代码在下面,下个数据集(我忘了在哪下的了…)、bert模型,把代码中的路径改下就好。这里只是用了bert的字向量,没有对bert做继续训练。
kashgari框架的代码更简单,有兴趣点这里。
首先,数据集样式:
海 O
钓 O
比 O
赛 O
地 O
点 O
在 O
厦 B-LOC
门 I-LOC
与 O
金 B-LOC
门 I-LOC
忘了在哪找的了,谁知道的话告诉我一下,我给贴上。
需要下载bert中文模型。
python版本3.7,安装包有两个你可能没有:
keras_bert
keras_contrib
keras_contrib安装教程:https://blog.csdn.net/qq_31456593/article/details/89388107
git安装教程:https://www.cnblogs.com/xueweisuoyong/p/11914045.html
代码都放下面了,也不多。
import numpy as np
from ke