SkyID
/
deepke
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix bug

This commit is contained in:
tlk-dsg 2021-10-21 15:32:08 +08:00
parent c71b173902
commit 74efd194bf
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/deepke/name_entity_re/standard/models/InferBert.py
View File

@ -135,7 +135,8 @@ class InferNer:
tmp.append(word)
if i==len(result)-1:
tmp.append(word)
if label=='B-PER' or label=='B-LOC' or label=='B-ORG':
tmp.append(word)
wordstype = result[i][1][2:]
tag[wordstype].append(''.join(tmp))