support confidences in find-entities

4 years ago · 49861b1652
parent 0da38d6ec6
commit 49861b1652
1 changed files with 1 additions and 1 deletions
--- a/tsvtools/tsv.py
+++ b/tsvtools/tsv.py
@ -15,7 +15,7 @@ def read_tsv(tsv_file):

 def write_tsv(tsv, urls, tsv_out_file):

-    if 'conf' in tsv.columns():
+    if 'conf' in tsv.columns:
        out_columns = ['No.', 'TOKEN', 'NE-TAG', 'NE-EMB', 'ID', 'url_id', 'left', 'right', 'top', 'bottom', 'conf']
    else:
        out_columns = ['No.', 'TOKEN', 'NE-TAG', 'NE-EMB', 'ID', 'url_id', 'left', 'right', 'top', 'bottom']