/tkitMarker_bert

使用bert微调提取实体,描述这些信息。

Primary LanguagePythonGNU General Public License v3.0GPL-3.0

tkitMarker_bert

tkitmarker_bert

安装

pip install tkitMarker_bert

测试

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

更多返回