qurator-spk/page2tsv
1
0
Fork 0
mirror of https://github.com/qurator-spk/page2tsv.git synced 2025-08-01 13:19:54 +02:00
Code Issues Releases Wiki Activity

enable NED only usage of find-entities

Browse source
This commit is contained in:
Kai 2020-05-25 15:10:08 +02:00
parent c12bea2cb0
commit 83fb2ea033
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
cli.py
View file

@ -336,7 +336,7 @@ def find_entities(tsv_file, tsv_out_file, ner_rest_endpoint, ned_rest_endpoint,
tmp = tsv.copy() tmp = tsv.copy()
tmp['sen'] = (tmp['No.'] == 0).cumsum() tmp['sen'] = (tmp['No.'] == 0).cumsum()
ner_result = [[{'word': row.TOKEN, 'prediction': row['NE-TAG']} for _, row in sen.iterrows] ner_result = [[{'word': row.TOKEN, 'prediction': row['NE-TAG']} for _, row in sen.iterrows()]
for _, sen in tmp.groupby('sen')] for _, sen in tmp.groupby('sen')]
else: else:
raise RuntimeError("Either NER rest endpoint or NER-TAG information within tsv_file required.") raise RuntimeError("Either NER rest endpoint or NER-TAG information within tsv_file required.")