文本标记tkitMarker_bert
{% embed url="https://github.com/napoler/tkitMarker_bert" %}
使用bert微调提取实体,描述这些信息。
安装
pip install tkitMarker_bert
测试
这里有个小模型
https://www.kaggle.com/terrychanorg/tmarkdescriptionmodel
from tkitMarker_bert import Marker
text="柯基犬是一个小短腿"
word="柯基犬"
#加载预测描述
pred=Marker(model_path="./model")
model,tokenizer=pred.load_model()
pall=pred.pre(word,text,model,tokenizer)
print(word,pall)
>>>柯基犬 ['是一个小短腿']
训练数据标记
https://github.com/napoler/tmark_Description